class ExecutionTestCase(unittest.TestCase): """ Test methods in Execution """ def __init__(self, *args, **kwargs): unittest.TestCase.__init__(self, *args, **kwargs) self.pkg_dir = os.path.join('..') self.execution = Execution(self.pkg_dir) self.img_size = 160 def test_create_wh(self): """ The warehouse should be a dictionary of UID keys and Person values :return: None """ directory = os.path.join(self.pkg_dir, 'dataset', 'test') warehouse = self.execution.create_wh(directory) condition = len(warehouse.get_samples()) > 0 self.assertEqual(condition, True) def test_acquire_data(self): """ This function should create two warehouses and populate them with samples :return: None """ self.execution.acquire_data() condition0 = len( self.execution.data_acquisition.trn_wh.get_samples()) > 0 condition1 = len( self.execution.data_acquisition.tst_wh.get_samples()) > 0 condition = condition0 and condition1 self.assertEqual(condition, True)