Extended test cases
@@ -73,15 +73,25 @@ class TestDot(unittest.TestCase):
@testing.parameterize(*testing.product({
'shape': [
- ((1, 2), (1, 2)),
- ((1, 3), (1, 3)),
- ((2, 2), (2, 2)),
- ((2, 3), (2, 3)),
- ((1, 2), (2, 2)),
- ((1, 3), (2, 3)),
- ((1, 2), (3, 2)),
- ((1, 3), (3, 3)),
- ]
+ ((2, 3, 4), (3, 4, 2)),
+ ((1, 1), (1, 1)),
+ ((1, 1), (1, 2)),
+ ((1, 2), (2, 1)),
+ ((2, 1), (1, 1)),
+ ((1, 2), (2, 3)),
+ ((2, 1), (1, 3)),
+ ((2, 3), (3, 1)),
+ ((2, 3), (3, 4)),
+ ((0, 3), (3, 4)),
+ ((2, 3), (3, 0)),
+ ((0, 3), (3, 0)),
+ ((3, 0), (0, 4)),
+ ((2, 3, 0), (3, 0, 2)),
+ ((0, 0), (0, 0)),
+ ((3,), (3,)),
+ ((2,), (2, 4)),
+ ((4, 2), (2,)),
+ ],
}))
@testing.gpu
class TestCrossProduct(unittest.TestCase):
