def pickle_modules(self, filename='plants.py'): file = [filename] parser = model.FileProcessor() parser.process_files(file) modules = parser.get_modules() pickler = pickle_modules.PickleModules() return pickler.save(modules)
def test_13_pickle_module(self): data_for_model = ['plants.py'] newModelData = model.FileProcessor() newModelData.process_files(data_for_model) model_data_module = newModelData.get_modules() pickler = pm.PickleModules() expected = len(model_data_module) actual = len(pickler.load()) self.assertEqual(expected, actual)
def load_pickle(self): ''' Loads previously saved module using pickle Author: Peter Command: load_pickle ''' pickler = pickle_modules.PickleModules() return pickler.load()
def test_14_pickle_module(self): data_for_model = ['plants.py'] newModelData = model.FileProcessor() newModelData.process_files(data_for_model) pickler = pm.PickleModules() actual = True try: expected = TypeError test = pickler.save() except TypeError: actual = TypeError self.assertEqual(expected, actual)
def do_pickle_modules(self, filename='plants.py'): ''' Load modules from single file and save them using pickle Author: Peter Command: pickle_modules filename eg pickle_modules plants.py ''' file = [filename] parser = model.FileProcessor() parser.process_files(file) modules = parser.get_modules() pickler = pickle_modules.PickleModules() return pickler.save(modules)
def test_10_pickle_save(self): pickler = pm.PickleModules() data = 12345 expected = True actual = pickler.save(data) self.assertEqual(expected, actual)
def test_11_pickle_load(self): pickler = pm.PickleModules() expected = 12345 actual = pickler.load() self.assertEqual(expected, actual)
def load_pickle(self): pickler = pickle_modules.PickleModules() return pickler.load()