class RecognizerSplitTest(unittest.TestCase):
def setUp(self):
self.original_base = ImageReader.base
ImageReader.base = 'tests/fixtures/' + ImageReader.base
self.subject = RecognizerSplit()
def tearDown(self):
ImageReader.base = self.original_base
def test_fit(self):
self.subject.load_data(['BLACK', 'BLUE'])
self.subject.fit()
def test_transform(self):
self.subject.load_data(['BLACK', 'BLUE'])
data = self.subject.transform()
self.assertEqual(len(data.X), len(data.y))
self.assertEqual(len(data.X), 12)
self.assertEqual(len(self.subject.test_data.X),
len(self.subject.test_data.y))
self.assertEqual(len(self.subject.test_data.X), 3)
class RecognizerSplitTest(unittest.TestCase): def setUp(self): self.original_base = ImageReader.base ImageReader.base = 'tests/fixtures/' + ImageReader.base self.subject = RecognizerSplit() def tearDown(self): ImageReader.base = self.original_base def test_fit(self): self.subject.load_data(['BLACK', 'BLUE']) self.subject.fit() def test_transform(self): self.subject.load_data(['BLACK', 'BLUE']) data = self.subject.transform() self.assertEqual(len(data.X), len(data.y)) self.assertEqual(len(data.X), 12) self.assertEqual( len(self.subject.test_data.X), len(self.subject.test_data.y)) self.assertEqual(len(self.subject.test_data.X), 3)
