def test_find_pattern_vertical_lines():
img = cv2.imread('images/pattern_rec_vert_lines.bmp')
pattern, pattern_color = pr.find_pattern(img)
assert pattern is Pattern.VERTICAL_LINES
assert pattern_color is Color.GREEN
def test_find_pattern_no_pattern():
img = cv2.imread('images/pattern_rec_no_pattern.bmp')
pattern, pattern_color = pr.find_pattern(img)
assert pattern is Pattern.NONE
assert pattern_color is Color.NONE
def test_find_pattern_inclined_grid():
img = cv2.imread('images/pattern_rec_inc_grid.bmp')
pattern, pattern_color = pr.find_pattern(img)
assert pattern is Pattern.INCLINED_GRID
assert pattern_color is Color.GREEN
def test_find_pattern_grid():
img = cv2.imread('images/pattern_rec_grid.bmp')
pattern, pattern_color = pr.find_pattern(img)
assert pattern is Pattern.GRID
assert pattern_color is Color.RED
def test_find_pattern_right_inclined_lines():
img = cv2.imread('images/pattern_rec_right_inc_lines.bmp')
pattern, pattern_color = pr.find_pattern(img)
assert pattern is Pattern.RIGHT_INCLINED_LINES
assert pattern_color is Color.YELLOW
def test_find_pattern_horizontal_lines():
img = cv2.imread('images/pattern_rec_hor_lines.bmp')
pattern, pattern_color = pr.find_pattern(img)
assert pattern is Pattern.HORIZONTAL_LINES
assert pattern_color is Color.RED