unit tests for #6315

Signed-off-by: Wenqi Li <[email protected]>
Project-MONAI · Apr 11, 2023 · 950a388 · 950a388
1 parent 768964b
commit 950a388
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/test_get_equivalent_dtype.py b/tests/test_get_equivalent_dtype.py
@@ -42,7 +42,15 @@ def test_native_type(self):
                 out_dtype = get_equivalent_dtype(n, type(im_dtype))
                 self.assertEqual(out_dtype, n)
 
-    @parameterized.expand([["float", np.float64], ["float32", np.float32], ["float64", np.float64]])
+    @parameterized.expand(
+        [
+            ["float", np.float64],
+            ["float32", np.float32],
+            ["np.float32", np.float32],
+            ["float64", np.float64],
+            ["torch.float64", np.float64],
+        ]
+    )
     def test_from_string(self, dtype_str, expected_np):
         expected_pt = get_equivalent_dtype(expected_np, torch.Tensor)
         # numpy