def test_no_range_error_when_in_bounds(self):
d = np.array([[[1]], [[1]], [[255]]])
try:
core.check_dataset_range(d)
except ValueError:
msg = "`check_dataset_range` raised ValueError on allowed color."
self.fail(msg)
def test_range_error_when_out_of_bounds(self):
with self.assertRaises(ValueError):
d = np.array([[[0]], [[-1]], [[0]]])
core.check_dataset_range(d)
with self.assertRaises(ValueError):
d = np.array([[[0]], [[256]], [[0]]])
core.check_dataset_range(d)