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()