Linting

2019-04-22 22:20:26 +02:00
parent 5cca608962
commit 1ce1d55705
2 changed files with 11 additions and 8 deletions
--- a/shenzhen_solitaire/cv/card_finder.py
+++ b/shenzhen_solitaire/cv/card_finder.py
@@ -102,8 +102,8 @@ def find_square(search_square: np.ndarray,
            best_square = square
            best_count = count
            best_coord = coord
-    assert best_square
-    assert best_coord
+    assert isinstance(best_square, np.ndarray)
+    assert isinstance(best_coord, tuple)
    cv2.imshow("Window", best_square -
               search_square[best_coord[0]:best_coord[0] +
                             best_square.shape[0], best_coord[1]:best_coord[1] +