From 582761a37ab1ebd04130d3eb1b5c7eb7f4b792b1 Mon Sep 17 00:00:00 2001
From: Ruben Young On <r.d.youngon@student.tudelft.nl>
Date: Mon, 27 May 2019 12:31:44 +0200
Subject: [PATCH] Fixed tests
---
tests/test_three_corners.py | 10 ++++------
1 file changed, 4 insertions(+), 6 deletions(-)
diff --git a/tests/test_three_corners.py b/tests/test_three_corners.py
index 1169be12..1e3fb938 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))
--
GitLab