def test_input(self): horizontal_flip = HorizontalFlip(image_in='x') output = horizontal_flip.forward(data=self.single_input, state={}) with self.subTest('Check output type'): self.assertEqual(type(output), list) with self.subTest('Check output image shape'): self.assertEqual(output[0].shape, self.single_output_shape)
def test_input_image_and_mask(self): horizontal_flip = HorizontalFlip(image_in='x', mask_in='x_mask') output = horizontal_flip.forward(data=self.input_image_and_mask, state={}) with self.subTest('Check output type'): self.assertEqual(type(output), list) with self.subTest('Check output image shape'): self.assertEqual(output[0].shape, self.image_and_mask_output_shape) with self.subTest('Check output mask shape'): self.assertEqual(output[1].shape, self.image_and_mask_output_shape)