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')