Update Resample test

fepegar · Sep 1, 2021 · 8eddf8a · 8eddf8a
1 parent 421e024
commit 8eddf8a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/transforms/preprocessing/test_resample.py b/tests/transforms/preprocessing/test_resample.py
@@ -74,7 +74,11 @@ def test_2d(self):
         self.assertEqual(shape, (1, 4, 6, 1))
 
     def test_input_list(self):
-        tio.Resample([1, 2, 3])
+        tio.Resample([1, 2, 3])(self.sample_subject)
+
+    def test_input_array(self):
+        resample = tio.Resample(np.asarray([1, 2, 3]))
+        resample(self.sample_subject)
 
     def test_image_target(self):
         tio.Resample(self.sample_subject.t1)(self.sample_subject)