def test_class_config(self):
'''Test class-specific configuration.'''
# Config section is not present
config = ConfigObj()
comp = Computation(config, None)
assert comp.myc == {}
# Config section contains 1 item
config = ConfigObj({'Computation': {
'test': 10,
}})
comp = Computation(config, None)
assert len(comp.myc) == 1
assert comp.myc['test'] == 10
def test_init(self):
'''Test basic object creation.'''
config = ConfigObj()
env = None
comp = Computation(config, env)
assert comp.config is config
assert comp.env is None
def test_get_fname(self):
'''Test file name generation.'''
# TODO: test keyword arguments
output_dir = 'test_output_dir'
fname_prefix = 'global_fname_prefix'
obj_fname_prefix = 'obj_fname_prefix'
filename = 'test_file.pdf'
config = ConfigObj({
'output_dir': output_dir,
'fname_prefix': fname_prefix,
'Computation': {
'fname_prefix': obj_fname_prefix
},
})
correct_path = os.path.join(
output_dir, '{fname_prefix}{obj_prefix}{file_base}'.format(
fname_prefix=fname_prefix,
obj_prefix=obj_fname_prefix,
file_base=filename))
comp = Computation(config, None)
assert comp.get_fname(filename) == correct_path
def test_get_fname(self):
'''Test file name generation.'''
# TODO: test keyword arguments
output_dir = 'test_output_dir'
fname_prefix = 'global_fname_prefix'
obj_fname_prefix = 'obj_fname_prefix'
filename = 'test_file.pdf'
config = ConfigObj({
'output_dir' : output_dir,
'fname_prefix' : fname_prefix,
'Computation' : {
'fname_prefix' : obj_fname_prefix
},
})
correct_path = os.path.join(
output_dir,
'{fname_prefix}{obj_prefix}{file_base}'.format(
fname_prefix=fname_prefix,
obj_prefix=obj_fname_prefix,
file_base=filename
)
)
comp = Computation(config, None)
assert comp.get_fname(filename) == correct_path
def test_abstract_methods(self):
'''Test abstract methods.'''
config = ConfigObj()
comp = Computation(config, None)
with pytest.raises(NotImplementedError):
comp.run_all()
def test_abstract_methods(self):
'''Test abstract methods.'''
config = ConfigObj()
comp = Computation(config, None)
with pytest.raises(NotImplementedError):
comp.run_all()
