Update test case for any/all with hardcode input matrix.

sighingnow · sighingnow · commit 560c524e7bb0 · 2018-01-13T14:19:23.000+08:00
Signed-off-by: HE, Tao &lt;sighingnow@gmail.com&gt;
diff --git a/test/test_torch.py b/test/test_torch.py
@@ -463,18 +463,20 @@ def test_all_any_empty(self):
 
     def test_all_any_with_dim(self):
         def test(x):
-            r1 = x.prod(dim=1, keepdim=False)
-            r2 = x.all(dim=1, keepdim=False)
+            r1 = x.prod(dim=0, keepdim=False)
+            r2 = x.all(dim=0, keepdim=False)
             self.assertEqual(r1.shape, r2.shape)
             self.assertTrue((r1 == r2).all())
 
-            r3 = x.sum(dim=2, keepdim=True).clamp(0, 1)
-            r4 = x.any(dim=2, keepdim=True)
+            r3 = x.sum(dim=1, keepdim=True).clamp(0, 1)
+            r4 = x.any(dim=1, keepdim=True)
             self.assertEqual(r3.shape, r4.shape)
             self.assertTrue((r3 == r4).all())
 
-        test(torch.rand((1, 2, 3, 4)).round().byte())
-        test(torch.rand((4, 3, 2, 1)).round().byte())
+        test(torch.ByteTensor([[0, 0, 0],
+                               [0, 0, 1],
+                               [0, 1, 1],
+                               [1, 1, 1]]))
 
     @unittest.skipIf(not torch.cuda.is_available(), 'no CUDA')
     def test_all_any_empty_cuda(self):
