def test_predict(self):
image = Image(filename=ImageReader('BLACK_BASE').filenames()[0]).image
ScreenReader.predict(image)
ScreenReader.predict(image)
ScreenReader.predict(image)
def test_DL_example(self):
img = ImageReader('BLACK_BASE').read()[0]
result = self.predict(img)
print(result)
result = self.predict(img)
print(result)
def setUp(self):
self.original_base = ImageReader.base
ImageReader.base = 'tests/fixtures/' + ImageReader.base
self.subject = ImageReader('BLACK')
