class TestGrounding(unittest.TestCase):
def setUp(self):
self.img = ImageFile('digits1')
self.img.remove_ground()
self.assertFalse(self.img.is_grounded())
self.segments = ContourSegmenter().process(self.img.image)
def test_textgrounder(self):
grounder = TextGrounder()
characters = "0" * len(self.segments)
grounder.ground(self.img, self.segments, characters)
self.assertTrue(self.img.is_grounded())
self.assertEquals(reconstruct_chars(self.img.ground.classes),
characters)
def test_textgrounder_wrong_len(self):
grounder = TextGrounder()
characters = "0" * len(self.segments)
with self.assertRaises(ValueError):
grounder.ground(self.img, self.segments, characters[:-4])
self.assertFalse(self.img.is_grounded())
def test_usergrounder(self):
ESC_KEY = 27
ZERO_KEY = 48
keys = [ZERO_KEY] * len(self.segments) + [ESC_KEY]
mock_generator = iter(keys)
def mock_input(*args):
return next(mock_generator)
grounder = UserGrounder()
with mock.patch('cv2.waitKey', mock_input):
with mock.patch('cv2.imshow'):
grounder.ground(self.img, self.segments)
self.assertTrue(self.img.is_grounded())
self.assertEquals(reconstruct_chars(self.img.ground.classes),
"0" * len(self.segments))
def test_terminal_grounder(self):
terminal = TerminalGrounder()
characters = "0" * len(self.segments)
mock_input_gen = iter(characters)
def mock_input(prompt):
return next(mock_input_gen)
with mock.patch('__builtin__.raw_input', mock_input):
terminal.ground(self.img, self.segments)
self.assertTrue(self.img.is_grounded())
self.assertEquals(reconstruct_chars(self.img.ground.classes),
"0" * len(self.segments))
def train(self, image_file):
"""feeds the training data to the OCR"""
if not isinstance(image_file, ImageFile):
image_file = ImageFile(image_file)
if not image_file.is_grounded():
raise Exception("The provided file is not grounded")
features = self.extractor.extract(image_file.image, image_file.ground.segments)
self.classifier.train(features, image_file.ground.classes)