fixes seeds

Signed-off-by: Wenqi Li <[email protected]>

Author: wyli

tests/test_rand_lambdad.py

@@ -15,7 +15,6 @@
 
 from monai.transforms import Randomizable
 from monai.transforms.utility.dictionary import RandLambdad
-from tests.utils import NumpyImageTestCase2D
 
 
 class RandTest(Randomizable):
@@ -24,26 +23,25 @@ class RandTest(Randomizable):
     """
 
     def randomize(self, data=None):
-        self.set_random_state(seed=134)
         self._a = self.R.random()
-        self.set_random_state(seed=None)
 
     def __call__(self, data):
         self.randomize()
         return data + self._a
 
 
-class TestRandLambdad(NumpyImageTestCase2D):
+class TestRandLambdad(unittest.TestCase):
     def test_rand_lambdad_identity(self):
-        img = self.imt
+        img = np.zeros((10, 10))
         data = {"img": img, "prop": 1.0}
 
         test_func = RandTest()
-
+        test_func.set_random_state(seed=134)
         expected = {"img": test_func(data["img"]), "prop": 1.0}
+        test_func.set_random_state(seed=134)
         ret = RandLambdad(keys=["img", "prop"], func=test_func, overwrite=[True, False])(data)
-        self.assertTrue(np.allclose(expected["img"], ret["img"]))
-        self.assertTrue(np.allclose(expected["prop"], ret["prop"]))
+        np.testing.assert_allclose(expected["img"], ret["img"])
+        np.testing.assert_allclose(expected["prop"], ret["prop"])
 
 
 if __name__ == "__main__":

tests/test_rand_rotate.py

@@ -52,7 +52,7 @@ def test_correct_results(self, degrees, keep_size, mode, padding_mode, align_cor
             self.imt[0, 0], -np.rad2deg(angle), (0, 1), not keep_size, order=_order, mode=_mode, prefilter=False
         )
         expected = np.stack(expected).astype(np.float32)
-        np.testing.assert_allclose(expected, rotated[0], rtol=1e-05)
+        np.testing.assert_allclose(expected, rotated[0])
 
 
 class TestRandRotate3D(NumpyImageTestCase3D):