def test_from_dict_with_invalid_cost_params(self):
with self.assertRaises(DataStoreError):
DataStoreConfig.from_dict({'description': 'Local files',
'title': 'Local',
'store_id': 'file',
'store_params': {'root': '.'},
'cost_params': {
# Required:
# 'input_pixels_per_punit': 10,
# 'output_pixels_per_punit': 20,
}})
def test_from_dict(self):
store_config = DataStoreConfig.from_dict({
'description': 'Local files',
'title': 'Local',
'store_id': 'file',
'store_params': {'root': '.'}
})
self.assertIsInstance(store_config, DataStoreConfig)
self.assertEqual('file', store_config.store_id)
self.assertEqual({'root': '.'}, store_config.store_params)
self.assertEqual('Local', store_config.title)
self.assertEqual('Local files', store_config.description)
def test_from_dict_with_valid_cost_params(self):
store_config = DataStoreConfig.from_dict({
'description': 'Local files',
'title': 'Local',
'store_id': 'file',
'store_params': {'root': '.'},
'cost_params': {
'input_pixels_per_punit': 500,
'output_pixels_per_punit': 100,
'input_punits_weight': 1.1,
}
})
self.assertIsInstance(store_config, DataStoreConfig)
self.assertEqual('file', store_config.store_id)
self.assertEqual({'root': '.'}, store_config.store_params)
self.assertEqual('Local', store_config.title)
self.assertEqual('Local files', store_config.description)