class ImageTest(unittest.TestCase): def setUp(self): self.subject = Image( filename='tests/fixtures/Training_Data/BLACK_BASE/' \ 'BLACK_16c25a59-c736-431f-8fe2-25664299eaea.png') def test_with_filename(self): self.assertTrue(self.subject.image is not None) def test_with_image(self): image2 = Image(image=self.subject.image) assert_equal(image2.image, self.subject.image) def test_feature(self): image2 = Image(image=self.subject.image) assert_equal(image2.feature(), self.subject.feature())
def test_predict(self): image = Image(filename=ImageReader('BLACK_BASE').filenames()[0]).image ScreenReader.predict(image) ScreenReader.predict(image) ScreenReader.predict(image)
def test_feature(self): image2 = Image(image=self.subject.image) assert_equal(image2.feature(), self.subject.feature())
def setUp(self): self.subject = Image( filename='tests/fixtures/Training_Data/BLACK_BASE/' \ 'BLACK_16c25a59-c736-431f-8fe2-25664299eaea.png')
def test_with_image(self): image2 = Image(image=self.subject.image) assert_equal(image2.image, self.subject.image)