def test_read_file_name(pdict_setup):
(prm_fd, prm_file_name) = tempfile.mkstemp()
prm_file = os.fdopen(prm_fd, "w")
prm_file.write(pdict_setup.param_dict_file)
prm_file.close()
param_dict = ModelParameterDictionary()
param_dict.read_from_file(prm_file_name)
os.remove(prm_file_name)
all_keys = set(
["FLOAT_VAL", "INT_VAL", "STRING_VAL", "TRUE_BOOL_VAL", "FALSE_BOOL_VAL"]
)
param_list = set(param_dict.params())
assert param_list == all_keys
def test_read_file_name():
(prm_fd, prm_file_name) = tempfile.mkstemp()
prm_file = os.fdopen(prm_fd, 'w')
prm_file.write(_TEST_PARAM_DICT_FILE)
prm_file.close()
param_dict = ModelParameterDictionary()
param_dict.read_from_file(prm_file_name)
os.remove(prm_file_name)
all_keys = set([
'FLOAT_VAL', 'INT_VAL', 'STRING_VAL', 'TRUE_BOOL_VAL',
'FALSE_BOOL_VAL',
])
param_list = set(param_dict.params())
assert_equal(param_list, all_keys)
def test_read_file_name(self):
(prm_fd, prm_file_name) = tempfile.mkstemp()
prm_file = os.fdopen(prm_fd, 'w')
prm_file.write(_TEST_PARAM_DICT_FILE)
prm_file.close()
param_dict = ModelParameterDictionary()
param_dict.read_from_file(prm_file_name)
os.remove(prm_file_name)
all_keys = set([
'FLOAT_VAL',
'INT_VAL',
'STRING_VAL',
'TRUE_BOOL_VAL',
'FALSE_BOOL_VAL',
])
param_list = set(param_dict.params())
self.assertEqual(param_list, all_keys)
class TestModelParameterDictionary(unittest.TestCase):
def setUp(self):
from StringIO import StringIO
self.param_file = StringIO(_TEST_PARAM_DICT_FILE)
self.param_dict = ModelParameterDictionary()
self.param_dict.read_from_file(self.param_file)
def test_read_file(self):
all_keys = set([
'FLOAT_VAL',
'INT_VAL',
'STRING_VAL',
'TRUE_BOOL_VAL',
'FALSE_BOOL_VAL',
])
param_list = set(self.param_dict.params())
self.assertEqual(param_list, all_keys)
def test_read_file_name(self):
(prm_fd, prm_file_name) = tempfile.mkstemp()
prm_file = os.fdopen(prm_fd, 'w')
prm_file.write(_TEST_PARAM_DICT_FILE)
prm_file.close()
param_dict = ModelParameterDictionary()
param_dict.read_from_file(prm_file_name)
os.remove(prm_file_name)
all_keys = set([
'FLOAT_VAL',
'INT_VAL',
'STRING_VAL',
'TRUE_BOOL_VAL',
'FALSE_BOOL_VAL',
])
param_list = set(param_dict.params())
self.assertEqual(param_list, all_keys)
def test_read_file_like_twice(self):
from StringIO import StringIO
param_file = StringIO(_TEST_PARAM_DICT_FILE)
param_dict_1 = ModelParameterDictionary()
param_dict_2 = ModelParameterDictionary()
param_dict_1.read_from_file(param_file)
param_dict_2.read_from_file(param_file)
def test_read_int(self):
self.assertEqual(self.param_dict.read_int('INT_VAL'), 1)
with self.assertRaises(ParameterValueError):
self.param_dict.read_int('FLOAT_VAL')
with self.assertRaises(MissingKeyError):
self.param_dict.read_int('MISSING_INT')
self.assertEqual(self.param_dict.read_int('MISSING_INT', 2), 2)
def test_get_int(self):
self.assertEqual(self.param_dict.get('INT_VAL', ptype=int), 1)
with self.assertRaises(ParameterValueError):
self.param_dict.get('FLOAT_VAL', ptype=int)
with self.assertRaises(MissingKeyError):
self.param_dict.get('MISSING_INT', ptype=int)
def test_set_default(self):
self.param_dict.setdefault('MISSING_INT', 2)
self.assertEqual(self.param_dict.read_int('MISSING_INT'), 2)
def test_read_float(self):
self.assertEqual(self.param_dict.read_float('FLOAT_VAL'), 2.2)
self.assertEqual(self.param_dict.read_float('INT_VAL'), 1)
with self.assertRaises(ParameterValueError):
self.param_dict.read_float('STRING_VAL')
with self.assertRaises(MissingKeyError):
self.param_dict.read_float('MISSING_FLOAT')
def test_read_string(self):
self.assertEqual(self.param_dict.read_string('STRING_VAL'),
'The Landlab')
self.assertEqual(self.param_dict.read_string('INT_VAL'), '1')
self.assertEqual(self.param_dict.read_string('FLOAT_VAL'), '2.2')
with self.assertRaises(MissingKeyError):
self.param_dict.read_string('MISSING_STRING')
def test_read_bool(self):
self.assertEqual(self.param_dict.read_bool('TRUE_BOOL_VAL'), True)
self.assertEqual(self.param_dict.read_bool('FALSE_BOOL_VAL'), False)
with self.assertRaises(MissingKeyError):
self.param_dict.read_bool('MISSING_BOOLEAN')
with self.assertRaises(ParameterValueError):
self.param_dict.read_bool('STRING_VAL')
def test_dict_keys(self):
all_keys = set([
'FLOAT_VAL',
'INT_VAL',
'STRING_VAL',
'TRUE_BOOL_VAL',
'FALSE_BOOL_VAL',
])
self.assertEqual(set(self.param_dict), all_keys)
for key in all_keys:
self.assertTrue(key in self.param_dict)
def test_dict_index(self):
self.assertEqual(self.param_dict['INT_VAL'], '1')
class TestModelParameterDictionary(unittest.TestCase):
def setUp(self):
from StringIO import StringIO
self.param_file = StringIO(_TEST_PARAM_DICT_FILE)
self.param_dict = ModelParameterDictionary()
self.param_dict.read_from_file(self.param_file)
def test_read_file(self):
all_keys = set([
'FLOAT_VAL', 'INT_VAL', 'STRING_VAL', 'TRUE_BOOL_VAL',
'FALSE_BOOL_VAL',
])
param_list = set(self.param_dict.params())
self.assertEqual(param_list, all_keys)
def test_read_file_name(self):
(prm_fd, prm_file_name) = tempfile.mkstemp()
prm_file = os.fdopen(prm_fd, 'w')
prm_file.write(_TEST_PARAM_DICT_FILE)
prm_file.close()
param_dict = ModelParameterDictionary()
param_dict.read_from_file(prm_file_name)
os.remove(prm_file_name)
all_keys = set([
'FLOAT_VAL', 'INT_VAL', 'STRING_VAL', 'TRUE_BOOL_VAL',
'FALSE_BOOL_VAL',
])
param_list = set(param_dict.params())
self.assertEqual(param_list, all_keys)
def test_read_file_like_twice(self):
from StringIO import StringIO
param_file = StringIO(_TEST_PARAM_DICT_FILE)
param_dict_1 = ModelParameterDictionary()
param_dict_2 = ModelParameterDictionary()
param_dict_1.read_from_file(param_file)
param_dict_2.read_from_file(param_file)
def test_read_int(self):
self.assertEqual(self.param_dict.read_int('INT_VAL'), 1)
with self.assertRaises(ParameterValueError):
self.param_dict.read_int('FLOAT_VAL')
with self.assertRaises(MissingKeyError):
self.param_dict.read_int('MISSING_INT')
self.assertEqual(self.param_dict.read_int('MISSING_INT', 2), 2)
def test_get_int(self):
self.assertEqual(self.param_dict.get('INT_VAL', ptype=int), 1)
with self.assertRaises(ParameterValueError):
self.param_dict.get('FLOAT_VAL', ptype=int)
with self.assertRaises(MissingKeyError):
self.param_dict.get('MISSING_INT', ptype=int)
def test_set_default(self):
self.param_dict.setdefault('MISSING_INT', 2)
self.assertEqual(self.param_dict.read_int('MISSING_INT'), 2)
def test_read_float(self):
self.assertEqual(self.param_dict.read_float('FLOAT_VAL'), 2.2)
self.assertEqual(self.param_dict.read_float('INT_VAL'), 1)
with self.assertRaises(ParameterValueError):
self.param_dict.read_float('STRING_VAL')
with self.assertRaises(MissingKeyError):
self.param_dict.read_float('MISSING_FLOAT')
def test_read_string(self):
self.assertEqual(self.param_dict.read_string('STRING_VAL'),
'The Landlab')
self.assertEqual(self.param_dict.read_string('INT_VAL'), '1')
self.assertEqual(self.param_dict.read_string('FLOAT_VAL'), '2.2')
with self.assertRaises(MissingKeyError):
self.param_dict.read_string('MISSING_STRING')
def test_read_bool(self):
self.assertEqual(self.param_dict.read_bool('TRUE_BOOL_VAL'), True)
self.assertEqual(self.param_dict.read_bool('FALSE_BOOL_VAL'), False)
with self.assertRaises(MissingKeyError):
self.param_dict.read_bool('MISSING_BOOLEAN')
with self.assertRaises(ParameterValueError):
self.param_dict.read_bool('STRING_VAL')
def test_dict_keys(self):
all_keys = set([
'FLOAT_VAL', 'INT_VAL', 'STRING_VAL', 'TRUE_BOOL_VAL',
'FALSE_BOOL_VAL',
])
self.assertEqual(set(self.param_dict), all_keys)
for key in all_keys:
self.assertTrue(key in self.param_dict)
def test_dict_index(self):
self.assertEqual(self.param_dict['INT_VAL'], '1')
