diff --git a/tests/test_three_corners.py b/tests/test_three_corners.py index 1169be12790279fffebab2502cbea72e01f7117f..1e3fb938e911061e2a5e741c619f9d86e7454a00 100644 --- a/tests/test_three_corners.py +++ b/tests/test_three_corners.py @@ -10,20 +10,18 @@ def test_three_straight_corners_1(): shape = (240, 200, 3) corner_markers = [(50, 50), (120, 50), (50, 200)] - top_left, corner_markers = fix_corner_markers(corner_markers, shape) + corner_markers = fix_corner_markers(corner_markers, shape) assert (120, 200) in corner_markers - assert top_left == (50, 50) def test_three_straight_corners_2(): shape = (240, 200, 3) corner_markers = [(120, 50), (50, 200), (120, 200)] - top_left, corner_markers = fix_corner_markers(corner_markers, shape) + corner_markers = fix_corner_markers(corner_markers, shape) assert (50, 50) in corner_markers - assert top_left == (50, 50) def test_pdf(datadir): @@ -48,8 +46,8 @@ def test_pdf(datadir): diff = [corner for corner in corners1 if corner not in corners2] diff_marker = min(diff) - _, fixed_corners2 = fix_corner_markers(corners2, page_img.shape) - added_marker = [corner for corner in fixed_corners2 if corner not in corners2][0] + fixed_corners2 = fix_corner_markers(corners2, page_img.shape) + added_marker = [corner for corner in fixed_corners2 if corner not in corners1][0] # Check if 'inferred' corner marker is not too far away dist = np.linalg.norm(np.subtract(added_marker, diff_marker))