(self, input_param, input_data, expected_value)
| 77 | class TestMapLabelValue(unittest.TestCase): |
| 78 | @parameterized.expand(TESTS) |
| 79 | def test_shape(self, input_param, input_data, expected_value): |
| 80 | result = MapLabelValue(**input_param)(input_data) |
| 81 | if isinstance(expected_value, torch.Tensor): |
| 82 | assert_allclose(result, expected_value) |
| 83 | else: |
| 84 | np.testing.assert_equal(result, expected_value) |
| 85 | self.assertTupleEqual(result.shape, expected_value.shape) |
| 86 | |
| 87 | |
| 88 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected