def test_set_pen_color_1(self):
drawing = Drawing(1920, 1080)
color = np.array([120, 150, 150])
drawing.set_pen_color(color)
color = cv2.cvtColor(np.array([[color]], dtype="uint8"), cv2.COLOR_BGR2HSV)[0, 0]
assert np.array_equal(drawing.pen_color_range[0], color - drawing.d_color_range)
assert np.array_equal(drawing.pen_color_range[1], color + drawing.d_color_range)
def test_set_pen_color_edge(self):
drawing = Drawing(1920, 1080)
color = np.array([0, 0, 0])
drawing.set_pen_color(color)
assert np.array_equal(drawing.pen_color_range[0], np.zeros(3))
assert np.array_equal(drawing.pen_color_range[1], color + drawing.d_color_range)
