Add test

fferflo · Jun 9, 2024 · ec14667 · ec14667
1 parent 19e875f
commit ec14667
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_shapes.py b/test/test_shapes.py
@@ -27,6 +27,7 @@ def test_shape_rearrange(test):
         2,
         2,
     )
+    assert einx.rearrange("b s... c x... -> x... b (s...) c", x, x=()).shape == (10, 400, 2)
 
     if backend.name != "torch":
         assert einx.rearrange("1 -> (x)", [1], x=10, backend=backend).shape == (10,)