Method test_keep_size

tests/transforms/test_rand_zoomd.py:72–83

(self)

            assert_allclose(zoomed[key], p(expected), atol=1.0, type_test=False)

    def test_keep_size(self):
        key = "img"
        random_zoom = RandZoomd(
            keys=key, prob=1.0, min_zoom=0.6, max_zoom=0.7, keep_size=True, padding_mode="constant", constant_values=2
        )
        for p in TEST_NDARRAYS_ALL:
            im = p(self.imt[0])
            zoomed = random_zoom({key: im})
            test_local_inversion(random_zoom, zoomed, {key: im}, key)
            np.testing.assert_array_equal(zoomed[key].shape, self.imt.shape[1:])
        random_zoom.prob = 0.0
        self.assertEqual(random_zoom({key: p(self.imt[0])})[key].dtype, torch.float32)

    @parameterized.expand(
        [("no_min_zoom", None, 1.1, "bilinear", TypeError), ("invalid_order", 0.9, 1.1, "s", ValueError)]