(self)
| 68 | assert_allclose(zoomed, p(expected), atol=1.0, type_test=False) |
| 69 | |
| 70 | def test_keep_size(self): |
| 71 | for p in TEST_NDARRAYS_ALL: |
| 72 | im = p(self.imt[0]) |
| 73 | random_zoom = RandZoom(prob=1.0, min_zoom=0.6, max_zoom=0.7, keep_size=True) |
| 74 | random_zoom.set_random_state(12) |
| 75 | zoomed = random_zoom(im) |
| 76 | test_local_inversion(random_zoom, zoomed, im) |
| 77 | self.assertTrue(np.array_equal(zoomed.shape, self.imt.shape[1:])) |
| 78 | zoomed = random_zoom(im) |
| 79 | self.assertTrue(np.array_equal(zoomed.shape, self.imt.shape[1:])) |
| 80 | zoomed = random_zoom(im) |
| 81 | self.assertTrue(np.array_equal(zoomed.shape, self.imt.shape[1:])) |
| 82 | random_zoom.prob = 0.0 |
| 83 | self.assertEqual(random_zoom(im).dtype, torch.float32) |
| 84 | |
| 85 | @parameterized.expand( |
| 86 | [("no_min_zoom", None, 1.1, "bilinear", TypeError), ("invalid_mode", 0.9, 1.1, "s", ValueError)] |
nothing calls this directly
no test coverage detected