def test_delete_lines_lines_none(self):
with pytest.raises(ValueError) as exc:
ld.delete_lines(self.test_image, None)
assert exc.value.message == "Lines can't be None."
def test_delete_lines_color_error(self):
with pytest.raises(ValueError) as exc:
ld.delete_lines(self.test_image, self.lines, color=(260, 0, 0))
assert exc.value.message == "Color value must be: (0-255, 0-255, 0-255)."
def test_delete_lines_good(self):
lines = ld.detect_lines(self.test_image)
r = ld.delete_lines(self.test_image, lines[0][0])
assert isinstance(r, np.ndarray)
def test_delete__lines_width_negative(self):
with pytest.raises(ValueError) as exc:
ld.delete_lines(self.test_image, self.lines, width=-10)
assert exc.value.message == "Width value must be greater than 0."