class ModuleTestDataExtraction(unittest.TestCase):
def setUp(self):
self._sudocube_factory = SudocubeFactory()
def test_should_detect_right_numbers_in_front_facing_image(self):
image = self._read_image(first_front_image_path)
sudocube = self._sudocube_factory.create_sudocube_from_image(image)
expected = first_front_image_expected
self._assert_same_unsolved_sudocube(expected, sudocube)
def test_should_detect_right_numbers_in_another_front_facing_image(self):
image = self._read_image(second_front_image_path)
sudocube = self._sudocube_factory.create_sudocube_from_image(image)
expected = second_front_image_expected
self._assert_same_unsolved_sudocube(expected, sudocube)
def _assert_same_unsolved_sudocube(self, expected, given):
data = given._data
self.assertEqual(expected["front"], data["front"])
self.assertEqual(expected["side"], data["side"])
self.assertEqual(expected["top"], data["top"])
self.assertEqual(expected["red square"], data["red square"])
def _read_image(self, path):
rgb = cv2.imread(path)
return rgb
class TestSudocubeFactory(unittest.TestCase):
@patch('cv2.GaussianBlur')
@patch('cv2.cvtColor')
def setUp(self, cv2_cvtColor_mock, cv2_GaussianBlur_mock):
#Constants
self._image = "AN IMAGE"
self._hsv_image = "AN HSV IMAGE"
self._smooth_image = "SMOOTH IMAGE"
self._unwarped_image = "UNWARPED IMAGE"
self._sudocube_frame = "SUDOCUBE FRAME"
self._green_frame = "GREEN FRAME"
self._corners = "CORNERS"
#Mocks
self._cv2_cvtColor_mock = cv2_cvtColor_mock
self._cv2_GaussianBlur_mock = cv2_GaussianBlur_mock
self._corner_finder_mock = MagicMock(name="corner_finder")
self._data_extractor_mock = MagicMock(name="data_extractor")
self._unwarper_mock = MagicMock(name="unwarper")
self._color_extractor_mock = MagicMock(name="color_extractor")
#Return value changes
self._cv2_cvtColor_mock.return_value = self._hsv_image
self._cv2_GaussianBlur_mock.return_value = self._smooth_image
self._corner_finder_mock.find_corners.return_value = self._corners
self._unwarper_mock.unwarp.return_value = self._unwarped_image
self._color_extractor_mock.extract_blue.return_value = self._sudocube_frame
self._color_extractor_mock.extract_green.return_value = self._green_frame
#Object creation
self._sudocube_factory = SudocubeFactory()
self._sudocube_factory._init_dependencies = self._fake_init_dependencies(self._sudocube_factory,
self._corner_finder_mock,
self._data_extractor_mock,
self._unwarper_mock,
self._color_extractor_mock)
#Execution
self._sudocube_factory.create_sudocube_from_image(self._image)
def test_should_convert_image_to_hsv(self):
self._cv2_cvtColor_mock.assert_called_once_with(self._image, cv2.COLOR_RGB2HSV)
def test_should_find_frame_of_sudocube_by_taking_only_its_color(self):
self._color_extractor_mock.extract_blue.assert_called_once_with(self._unwarped_image)
def test_should_tell_corner_finder_to_find_outside_corners_of_sudocube(self):
self._corner_finder_mock.find_corners.assert_called_once_with(self._sudocube_frame)
def test_should_tell_sudocube_data_extractor_to_find_data_from_image_and_corner_positions(self):
self._data_extractor_mock.extract_data_from_image_with_corner_info.assert_called_once_with(self._unwarped_image,
self._corners)
#Monkey patch to mock dependencies
def _fake_init_dependencies(self, sudocube_factory, corner_finder, data_extractor, unwarper, color_extractor):
sudocube_factory._corner_finder = corner_finder
sudocube_factory._data_extractor = data_extractor
sudocube_factory._unwarper = unwarper
sudocube_factory._color_extractor = color_extractor
def setUp(self, cv2_cvtColor_mock, cv2_GaussianBlur_mock):
#Constants
self._image = "AN IMAGE"
self._hsv_image = "AN HSV IMAGE"
self._smooth_image = "SMOOTH IMAGE"
self._unwarped_image = "UNWARPED IMAGE"
self._sudocube_frame = "SUDOCUBE FRAME"
self._green_frame = "GREEN FRAME"
self._corners = "CORNERS"
#Mocks
self._cv2_cvtColor_mock = cv2_cvtColor_mock
self._cv2_GaussianBlur_mock = cv2_GaussianBlur_mock
self._corner_finder_mock = MagicMock(name="corner_finder")
self._data_extractor_mock = MagicMock(name="data_extractor")
self._unwarper_mock = MagicMock(name="unwarper")
self._color_extractor_mock = MagicMock(name="color_extractor")
#Return value changes
self._cv2_cvtColor_mock.return_value = self._hsv_image
self._cv2_GaussianBlur_mock.return_value = self._smooth_image
self._corner_finder_mock.find_corners.return_value = self._corners
self._unwarper_mock.unwarp.return_value = self._unwarped_image
self._color_extractor_mock.extract_blue.return_value = self._sudocube_frame
self._color_extractor_mock.extract_green.return_value = self._green_frame
#Object creation
self._sudocube_factory = SudocubeFactory()
self._sudocube_factory._init_dependencies = self._fake_init_dependencies(self._sudocube_factory,
self._corner_finder_mock,
self._data_extractor_mock,
self._unwarper_mock,
self._color_extractor_mock)
#Execution
self._sudocube_factory.create_sudocube_from_image(self._image)
def setUp(self):
self._sudocube_factory = SudocubeFactory()
