Method test_shape

tests/transforms/test_repeat_channeld.py:35–37

(self, input_param, input_data, expected_shape)

class TestRepeatChanneld(unittest.TestCase):
    @parameterized.expand(TESTS)
    def test_shape(self, input_param, input_data, expected_shape):
        result = RepeatChanneld(**input_param)(input_data)
        self.assertEqual(result["img"].shape, expected_shape)


if __name__ == "__main__":