Add compound tests

Signed-off-by: Tuan Chien <[email protected]>

Author: tuanchien

tests/test_some_of.py

@@ -13,9 +13,14 @@
 
 import unittest
 
-from monai.transforms import Transform
-from monai.transforms.compose import SomeOf
+from parameterized import parameterized
+
+from monai.data import MetaTensor
+from monai.transforms import SomeOf, TraceableTransform, Transform
+from monai.transforms.compose import Compose, SomeOf
 from monai.utils import set_determinism
+from monai.utils.enums import TraceKeys
+from tests.test_one_of import A, B, C, Inv, NonInv, X, Y
 
 
 class A(Transform):
@@ -33,6 +38,19 @@ def __call__(self, x):
         return 5 * x
 
 
+class D(Transform):
+    def __call__(self, x):
+        return 7 * x
+
+
+KEYS = ["x", "y"]
+TEST_COMPOUND = [
+    (SomeOf((A(), B(), C()), fixed=True, max_num_transforms=3), 2 * 3 * 5),
+    (Compose((SomeOf((A(), B(), C()), fixed=True, max_num_transforms=3), D())), 2 * 3 * 5 * 7),
+    (SomeOf((A(), B(), C(), Compose((D()))), fixed=True, max_num_transforms=4), 2 * 3 * 5 * 7),
+]
+
+
 class TestSomeOf(unittest.TestCase):
     def setUp(self):
         set_determinism(seed=0)
@@ -90,6 +108,11 @@ def test_unfixed(self):
         for i in range(4):
             self.assertAlmostEqual(subset_size_counts[i] / iterations, 0.25, delta=0.01)
 
+    @parameterized.expand(TEST_COMPOUND)
+    def test_compound_pipeline(self, transform, expected_value):
+        output = transform(1)
+        self.assertEqual(output, expected_value)
+
 
 if __name__ == "__main__":
     unittest.main()