combining split tests

Author: apbose

tests/py/dynamo/converters/test_split_aten.py

@@ -33,6 +33,7 @@ def forward(self, input):
     @parameterized.expand(
         [
             ("split_size_or_sections_list_no_dim_list", [1, 4]),
+            ("split_size_or_sections_list_no_dim_not_full_list", [1, 3]),
         ]
     )
     def test_split_list(self, _, split_size_or_tensor):
@@ -52,29 +53,6 @@ def forward(self, input):
             disable_passes=True,
         )
 
-    @parameterized.expand(
-        [
-            ("split_size_or_sections_list_no_dim_not_full_list", [1, 3]),
-        ]
-    )
-    def test_split_not_full_list(self, _, split_size_or_tensor):
-        class TestModule(torch.nn.Module):
-            def __init__(self):
-                super().__init__()
-
-            def forward(self, input):
-                out = torch.split(input, split_size_or_tensor)
-                return out
-
-        input = [torch.randn(10).reshape(5, 2)]
-        with self.assertRaises(RuntimeError):
-            self.run_test(
-                TestModule(),
-                input,
-                expected_ops={torch.ops.aten.split_with_sizes.default},
-                disable_passes=True,
-            )
-
     @parameterized.expand(
         [
             ("split_size_or_sections_dims", 2, 1),
@@ -97,28 +75,6 @@ def forward(self, input):
             disable_passes=True,
         )
 
-    @parameterized.expand(
-        [
-            ("split_size_or_sections_list_dims", [1, 1], 1),
-        ]
-    )
-    def test_split_dim(self, _, split_size_or_tensor, dim):
-        class TestModule(torch.nn.Module):
-            def __init__(self):
-                super().__init__()
-
-            def forward(self, input):
-                out = torch.split(input, split_size_or_tensor, dim)
-                return out
-
-        input = [torch.randn(10).reshape(5, 2)]
-        self.run_test(
-            TestModule(),
-            input,
-            expected_ops={torch.ops.aten.split_with_sizes.default},
-            disable_passes=True,
-        )
-
     @parameterized.expand(
         [
             ("split_size_or_sections_list_dims", [1, 1], 1),