()
| 39 | |
| 40 | |
| 41 | def test_ColorJitter(): |
| 42 | t = ColorJitter() |
| 43 | aug_data = t.apply_batch(generate_data()) |
| 44 | aug_data_shape = [(a.shape, b.shape) for a, b in aug_data] |
| 45 | target_shape = [(data_shape, label_shape)] * 4 |
| 46 | assert aug_data_shape == target_shape |
| 47 | |
| 48 | |
| 49 | def test_RandomHorizontalFlip(): |
nothing calls this directly
no test coverage detected