Method test_shape

tests/transforms/test_repeat_channel.py:28–30

(self, input_param, input_data, expected_shape)

class TestRepeatChannel(unittest.TestCase):
    @parameterized.expand(TESTS)
    def test_shape(self, input_param, input_data, expected_shape):
        result = RepeatChannel(**input_param)(input_data)
        self.assertEqual(result.shape, expected_shape)


if __name__ == "__main__":