(self, input_param, test_data, expected_shape)
| 82 | class TestSqueezeDim(unittest.TestCase): |
| 83 | @parameterized.expand(TESTS) |
| 84 | def test_shape(self, input_param, test_data, expected_shape): |
| 85 | result = SqueezeDimd(**input_param)(test_data) |
| 86 | self.assertTupleEqual(result["img"].shape, expected_shape) |
| 87 | self.assertTupleEqual(result["seg"].shape, expected_shape) |
| 88 | if "dim" in input_param and isinstance(result["img"], MetaTensor) and input_param["dim"] == 2: |
| 89 | assert_allclose(result["img"].affine.shape, [3, 3]) |
| 90 | |
| 91 | @parameterized.expand(TESTS_FAIL) |
| 92 | def test_invalid_inputs(self, exception, input_param, test_data): |
nothing calls this directly
no test coverage detected