def test_custom_kw_config_define_and_read(self):
data_1 = {"VALUE_1": 123453.3,
"VALUE_2": 0.234234}
data_2 = {"VALUE_1": 965689,
"VALUE_3": 1.1222}
with TestAreaContext("python/enkf/data/custom_kw_config") as test_area:
self.createResultFile("result_file_1", data_1)
self.createResultFile("result_file_2", data_2)
custom_kw_config = CustomKWConfig("CUSTOM_KW", "result_file")
result_1 = StringList()
success = custom_kw_config.parseResultFile("result_file_1", result_1)
self.assertTrue(success)
result_2 = StringList()
success = custom_kw_config.parseResultFile("result_file_2", result_2)
self.assertFalse(success)
for key in custom_kw_config:
self.assertTrue(key in data_1)
self.assertFalse("VALUE_3" in custom_kw_config)
def test_custom_kw_config_parse_fail(self):
data = {"KEY_1": "Value Key_2"}
with TestAreaContext("python/enkf/data/custom_kw_config_fail") as test_area:
self.createResultFile("result_file", data)
custom_kw_config = CustomKWConfig("CUSTOM_KW_FAIL", "result_file")
self.assertIsNone(custom_kw_config.getOutputFile())
self.assertFalse(custom_kw_config.parseResultFile("result_file", StringList()))
def test_custom_kw_config_construction_with_definition(self):
definition = {
"VALUE_1": float,
"VALUE_2": str
}
custom_kw_config = CustomKWConfig("TEST", None, definition=definition)
self.assertEqual(len(custom_kw_config.getKeys()), 2)
self.assertTrue(custom_kw_config.keyIsDouble("VALUE_1"))
self.assertFalse(custom_kw_config.keyIsDouble("VALUE_2"))
self.assertIn("VALUE_1", custom_kw_config)
self.assertIn("VALUE_2", custom_kw_config)
self.assertNotIn("VALUE_3", custom_kw_config)
def test_custom_kw_config_multiple_identical_keys(self):
data = {"VALUE_1": 2345.234,
"VALUE_2": 0.001234,
"VALUE_3": "string_1",
"VALUE_4": "string_2 VALUE_4 repeat_of_value_4"}
with TestAreaContext("python/enkf/data/custom_kw_config_multiple_identical_keys") as test_area:
self.createResultFile("result_file", data)
custom_kw_config = CustomKWConfig("CUSTOM_KW", "result_file")
result = StringList()
success = custom_kw_config.parseResultFile("result_file", result)
self.assertTrue(success)
index_of_value_4 = custom_kw_config.indexOfKey("VALUE_4")
self.assertEqual(result[index_of_value_4], "repeat_of_value_4")
def test_custom_kw_config_creation(self):
data = {"VALUE_1": 2345.234,
"VALUE_2": 0.001234,
"VALUE_3": "string_1",
"VALUE_4": "string_2"}
with TestAreaContext("python/enkf/data/custom_kw_config") as test_area:
self.createResultFile("result_file", data)
custom_kw_config = CustomKWConfig("CUSTOM_KW", "result_file", "output_file")
self.assertEqual(custom_kw_config.getName(), "CUSTOM_KW")
self.assertEqual(custom_kw_config.getResultFile(), "result_file")
self.assertEqual(custom_kw_config.getOutputFile(), "output_file")
self.assertEqual(len(custom_kw_config), 0)
result = StringList()
success = custom_kw_config.parseResultFile("result_file", result)
self.assertTrue(success)
self.assertEqual(len(custom_kw_config), 4)
for index, key in enumerate(data):
self.assertTrue(key in custom_kw_config)
key_is_string = isinstance(data[key], str)
self.assertTrue(custom_kw_config.keyIsDouble(key) != key_is_string)
self.assertEqual(index, custom_kw_config.indexOfKey(key))
self.assertEqual(result[index], str(data[key]))
self.assertTrue(len(custom_kw_config.getKeys()) == 4)
for key in custom_kw_config:
self.assertTrue(key in data)
