def test_12_pickle_module(self):
data_for_model = ['plants.py']
newModelData = model.FileProcessor()
newModelData.process_files(data_for_model)
model_data_module = newModelData.get_modules()
pickler = PickleModules()
expected = True
actual = pickler.save(model_data_module)
self.assertEqual(expected, actual)
def test_14_pickle_module(self):
data_for_model = ['plants.py']
newModelData = model.FileProcessor()
newModelData.process_files(data_for_model)
pickler = 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 = PickleModules()
return pickler.save(modules)
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 = FileProcessor(self.controller.statistics)
parser.process_files(file)
modules = parser.get_modules()
pickler = PickleModules()
return pickler.save(modules)
def test_10_pickle_save(self):
pickler = PickleModules()
data = 12345
expected = True
actual = pickler.save(data)
self.assertEqual(expected, actual)