class ImageReaderTest(unittest.TestCase):
def setUp(self):
self.original_base = ImageReader.base
ImageReader.base = 'tests/fixtures/' + ImageReader.base
self.subject = ImageReader('BLACK')
def tearDown(self):
ImageReader.base = self.original_base
def test_path_of(self):
self.assertEqual(self.subject.path_of(),
'tests/fixtures/Training_Data/*BLACK*/*.png')
def test_filenames(self):
self.assertEqual(len(self.subject.filenames()), 10)
def test_read(self):
for feature in self.subject.read():
self.assertTrue(feature is not None)
class ImageReaderTest(unittest.TestCase):
def setUp(self):
self.original_base = ImageReader.base
ImageReader.base = 'tests/fixtures/' + ImageReader.base
self.subject = ImageReader('BLACK')
def tearDown(self):
ImageReader.base = self.original_base
def test_path_of(self):
self.assertEqual(
self.subject.path_of(),
'tests/fixtures/Training_Data/*BLACK*/*.png')
def test_filenames(self):
self.assertEqual(len(self.subject.filenames()), 10)
def test_read(self):
for feature in self.subject.read():
self.assertTrue(feature is not None)
def test_DL_example(self):
img = ImageReader('BLACK_BASE').read()[0]
result = self.predict(img)
print(result)
result = self.predict(img)
print(result)
def test_predict(self):
image = Image(filename=ImageReader('BLACK_BASE').filenames()[0]).image
ScreenReader.predict(image)
ScreenReader.predict(image)
ScreenReader.predict(image)
def setUp(self):
self.original_base = ImageReader.base
ImageReader.base = 'tests/fixtures/' + ImageReader.base
self.subject = ImageReader('BLACK')
def setUp(self):
self.original_base = ImageReader.base
ImageReader.base = 'tests/fixtures/' + ImageReader.base
self.subject = ImageReader('BLACK')
