fixes tests

Signed-off-by: Wenqi Li <wenqil@nvidia.com>

Author: wyli

tests/test_compose.py

@@ -169,16 +169,16 @@ def test_data_loader(self):
         set_determinism(seed=123)
         train_loader = DataLoader(train_ds, num_workers=0)
         out_1 = next(iter(train_loader))
-        self.assertAlmostEqual(out_1.cpu().item(), 0.84291356)
+        self.assertAlmostEqual(out_1.cpu().item(), 0.0409280)
 
         if sys.platform != "win32":  # skip multi-worker tests on win32
             train_loader = DataLoader(train_ds, num_workers=1)
             out_1 = next(iter(train_loader))
-            self.assertAlmostEqual(out_1.cpu().item(), 0.180814653)
+            self.assertAlmostEqual(out_1.cpu().item(), 0.78663897075)
 
             train_loader = DataLoader(train_ds, num_workers=2)
             out_1 = next(iter(train_loader))
-            self.assertAlmostEqual(out_1.cpu().item(), 0.04293707)
+            self.assertAlmostEqual(out_1.cpu().item(), 0.785907334)
         set_determinism(None)
 
     def test_data_loader_2(self):
@@ -191,16 +191,16 @@ def test_data_loader_2(self):
 
         train_loader = DataLoader(train_ds, num_workers=0)
         out_2 = next(iter(train_loader))
-        self.assertAlmostEqual(out_2.cpu().item(), 0.7858843729)
+        self.assertAlmostEqual(out_2.cpu().item(), 0.98921915918)
 
         if sys.platform != "win32":  # skip multi-worker tests on win32
             train_loader = DataLoader(train_ds, num_workers=1)
             out_2 = next(iter(train_loader))
-            self.assertAlmostEqual(out_2.cpu().item(), 0.305763411)
+            self.assertAlmostEqual(out_2.cpu().item(), 0.32985207)
 
             train_loader = DataLoader(train_ds, num_workers=2)
             out_1 = next(iter(train_loader))
-            self.assertAlmostEqual(out_1.cpu().item(), 0.131966779)
+            self.assertAlmostEqual(out_1.cpu().item(), 0.28602141572)
         set_determinism(None)
 
     def test_flatten_and_len(self):

tests/test_grid_dataset.py

@@ -60,7 +60,7 @@ def test_loading_array(self):
             np.testing.assert_equal(tuple(item[0].shape), (2, 1, 2, 2))
         np.testing.assert_allclose(
             item[0],
-            np.array([[[[8.0577, 9.0577], [12.0577, 13.0577]]], [[[10.5540, 11.5540], [14.5540, 15.5540]]]]),
+            np.array([[[[8.240326, 9.240326], [12.240326, 13.240326]]], [[[10.1624, 11.1624], [14.1624, 15.1624]]]]),
             rtol=1e-4,
         )
         np.testing.assert_allclose(item[1], np.array([[[0, 1], [2, 4], [0, 2]], [[0, 1], [2, 4], [2, 4]]]), rtol=1e-5)
@@ -69,7 +69,9 @@ def test_loading_array(self):
                 np.testing.assert_equal(tuple(item[0].shape), (2, 1, 2, 2))
             np.testing.assert_allclose(
                 item[0],
-                np.array([[[[7.6533, 8.6533], [11.6533, 12.6533]]], [[[9.8524, 10.8524], [13.8524, 14.8524]]]]),
+                np.array(
+                    [[[[7.723618, 8.723618], [11.723618, 12.723618]]], [[[10.7175, 11.7175], [14.7175, 15.7175]]]]
+                ),
                 rtol=1e-3,
             )
             np.testing.assert_allclose(
@@ -102,7 +104,7 @@ def test_loading_dict(self):
             self.assertListEqual(item[0]["metadata"], ["test string", "test string"])
         np.testing.assert_allclose(
             item[0]["image"],
-            np.array([[[[8.0577, 9.0577], [12.0577, 13.0577]]], [[[10.5540, 11.5540], [14.5540, 15.5540]]]]),
+            np.array([[[[8.240326, 9.240326], [12.240326, 13.240326]]], [[[10.1624, 11.1624], [14.1624, 15.1624]]]]),
             rtol=1e-4,
         )
         np.testing.assert_allclose(item[1], np.array([[[0, 1], [2, 4], [0, 2]], [[0, 1], [2, 4], [2, 4]]]), rtol=1e-5)
@@ -111,7 +113,9 @@ def test_loading_dict(self):
                 np.testing.assert_equal(item[0]["image"].shape, (2, 1, 2, 2))
             np.testing.assert_allclose(
                 item[0]["image"],
-                np.array([[[[7.6533, 8.6533], [11.6533, 12.6533]]], [[[9.8524, 10.8524], [13.8524, 14.8524]]]]),
+                np.array(
+                    [[[[7.723618, 8.723618], [11.723618, 12.723618]]], [[[10.7175, 11.7175], [14.7175, 15.7175]]]]
+                ),
                 rtol=1e-3,
             )
             np.testing.assert_allclose(

tests/test_patch_dataset.py

@@ -59,7 +59,7 @@ def test_loading_array(self):
         np.testing.assert_allclose(
             item[0],
             np.array(
-                [[[1.338681, 2.338681, 3.338681], [5.338681, 6.338681, 7.338681], [9.338681, 10.338681, 11.338681]]]
+                [[[-0.593095, 0.406905, 1.406905], [3.406905, 4.406905, 5.406905], [7.406905, 8.406905, 9.406905]]]
             ),
             rtol=1e-5,
         )
@@ -69,13 +69,7 @@ def test_loading_array(self):
             np.testing.assert_allclose(
                 item[0],
                 np.array(
-                    [
-                        [
-                            [4.957847, 5.957847, 6.957847],
-                            [8.957847, 9.957847, 10.957847],
-                            [12.957847, 13.957847, 14.957847],
-                        ]
-                    ]
+                    [[[0.234308, 1.234308, 2.234308], [4.234308, 5.234308, 6.234308], [8.234308, 9.234308, 10.234308]]]
                 ),
                 rtol=1e-5,
             )