def test_merge_launchers_with_definitions(self):
local_config = {'models': [{
'name': 'model',
'launchers': [{'framework': 'dlsdk'}],
'datasets': [{'name': 'global_dataset'}]
}]}
config = ConfigReader._merge_configs(self.global_config, local_config)
assert config['models'][0]['launchers'][0] == self.get_global_launcher('dlsdk')
def test_merge_datasets_with_definitions(self):
local_config = {'models': [{
'name': 'model',
'launchers': [{'framework': 'dlsdk', 'model': '/absolute_path', 'weights': '/absolute_path'}],
'datasets': [{'name': 'global_dataset'}]
}]}
config = ConfigReader._merge_configs(self.global_config, local_config)
assert config['models'][0]['datasets'][0] == self.global_datasets[0]
def test_merge_launchers_with_model_is_not_modified(self):
local_config = {'models': [{
'name': 'model',
'launchers': [{'framework': 'dlsdk', 'model': 'custom'}],
'datasets': [{'name': 'global_dataset'}]
}]}
expected = self.get_global_launcher('dlsdk')
expected['model'] = 'custom'
config = ConfigReader._merge_configs(self.global_config, local_config)
assert config['models'][0]['launchers'][0] == expected
def test_merge_datasets_with_definitions_and_meta_is_not_modified(self):
local_config = {'models': [{
'name': 'model',
'launchers': [{'framework': 'dlsdk', 'model': '/absolute_path', 'weights': '/absolute_path'}],
'datasets': [{'name': 'global_dataset', 'dataset_meta': '/absolute_path'}]
}]}
expected = self.global_datasets[0]
expected['dataset_meta'] = '/absolute_path'
config = ConfigReader._merge_configs(self.global_config, local_config)
assert config['models'][0]['datasets'][0] == expected
