()
| 23 | |
| 24 | |
| 25 | def test_ToMode(): |
| 26 | t = ToMode(mode="CHW") |
| 27 | aug_data = t.apply_batch(generate_data()) |
| 28 | aug_data_shape = [(a.shape, b.shape) for a, b in aug_data] |
| 29 | target_shape = [(ToMode_target_shape, label_shape)] * 4 |
| 30 | assert aug_data_shape == target_shape |
| 31 | |
| 32 | |
| 33 | def test_CenterCrop(): |
nothing calls this directly
no test coverage detected