(self, in_type, input_param, expected_shape)
| 29 | class TestAsChannelLast(unittest.TestCase): |
| 30 | @parameterized.expand(TESTS) |
| 31 | def test_shape(self, in_type, input_param, expected_shape): |
| 32 | test_data = in_type(np.random.randint(0, 2, size=[1, 2, 3, 4])) |
| 33 | result = AsChannelLast(**input_param)(test_data) |
| 34 | self.assertTupleEqual(result.shape, expected_shape) |
| 35 | |
| 36 | |
| 37 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected