def test_random_crop_7(self, mocker):
random_crop = RandomCrop((30, 100))
image = np.zeros((1, 100, 100))
mask = np.zeros((100, 100))
result = random_crop({"image": image, "mask": mask})
assert result['image'].shape == (1, 30, 100)
assert result['mask'].shape == (30, 100)
def test_random_crop_5(self, mocker):
random_crop = RandomCrop((30, 60))
image = np.zeros((3, 100, 100))
image[2, 0, 0] = 1
mask = np.zeros((100, 100))
mask[0, 0] = 1
result = random_crop({"image": image, "mask": mask})
assert result['image'].shape == (3, 30, 60)
assert result['mask'].shape == (30, 60)
def test_random_crop(self, mocker):
mocker.patch.object(np.random, "randint", return_value=0)
random_crop = RandomCrop(100)
image = np.zeros((3, 200, 100))
image[0, 0, 0] = 1
mask = np.zeros((200, 100))
mask[0, 0] = 1
result = random_crop({"image": image, "mask": mask})
assert result['image'].shape == (3, 100, 100)
assert result['mask'].shape == (100, 100)
assert result['image'][0, 0, 0] == 1
assert result['mask'][0, 0] == 1