lukas/shenzhen-solitaire
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Made border detection work

Browse Source
This commit is contained in:
Lukas Wölfer
2020-02-08 00:42:55 +01:00
parent cf89e4c694
commit 07f8cae238
7 changed files with 112 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
test/test_cv.py
View File

@@ -7,7 +7,7 @@ import numpy as np
from shenzhen_solitaire import board
from shenzhen_solitaire.card_detection import adjustment, board_parser
from shenzhen_solitaire.card_detection.configuration import Configuration
import shenzhen_solitaire.card_detection.configuration as configuration
from . import boards
@@ -16,10 +16,8 @@ class CardDetectionTest(unittest.TestCase):
"""Parse a configuration and a board"""
image = cv2.imread("pictures/20190809172206_1.jpg")
loaded_config = Configuration.load("test_config.zip")
loaded_config = configuration.load("test_config.zip")
my_board = board_parser.parse_board(image, loaded_config)
for rows in zip(boards.B20190809172206_1.field, my_board.field):
for good_cell, test_cell in zip(*rows):
self.assertEqual(good_cell, test_cell)
for correct_row, my_row in zip(boards.B20190809172206_1.field, my_board.field):
self.assertListEqual(correct_row, my_row)