class TestOCR(unittest.TestCase):
def setUp(self):
#TODO: test axes, test one_hot
try:
self.train = OCR(which_set='train')
self.test = OCR(which_set='test')
except (NoDataPathError, NotInstalledError):
raise SkipTest()
def get_test_set(self):
try:
test_from_train = self.train.get_test_set()
self.assertTrue(numpy.all(test_from_train.get_design_matrix() == self.test.get_design_matrix()))
self.assertTrue(numpy.all(test_from_train.get_targets() == self.test.get_targets()))
except (NoDataPathError, NotInstalledError):
raise SkipTest()
