def test_no_shape_error_when_4d_numpy_array(self):
one_red_pixel = np.array([[[255]], [[0]], [[0]]])
one_green_pixel = np.array([[[0]], [[255]], [[0]]])
d = np.array([one_red_pixel, one_green_pixel])
try:
core.check_dataset(d)
except ValueError as e:
msg = "`check_dataset` ValueError on 4D numpy array.\n{}"
self.fail(msg.format(e))
def test_fix_shape_error_when_PIL_format(self):
one_green_pixel = np.array([[[0]], [[255]], [[0]]])
pil_green_pixel = np.array([[[0, 255, 0]]])
with self.assertRaises(ValueError):
core.check_dataset(pil_green_pixel)
d = core.try_fix_dataset(pil_green_pixel)
try:
core.check_dataset(d)
except ValueError as e:
msg = "`check_dataset` ValueError on fixed dataset.\n{}"
self.fail(msg.format(e))
self.assertEqual(True, (d == one_green_pixel).all())
def test_fix_shape_error_when_PIL_format_in_4d_numpy_array(self):
red_pixel = np.array([[[255]], [[0]], [[0]]])
green_pixel = np.array([[[0]], [[255]], [[0]]])
d = np.array([red_pixel, green_pixel])
pil_d = np.array([red_pixel.transpose(), green_pixel.transpose()])
with self.assertRaises(ValueError):
core.check_dataset(pil_d)
fixed_d = core.try_fix_dataset(pil_d)
try:
core.check_dataset(fixed_d)
except ValueError as e:
msg = "`check_dataset` ValueError on fixed 4D dataset.\n{}"
self.fail(msg.format(e))
self.assertEqual(True, (fixed_d == d).all())