def test_load_conf_os_env(self, mock_args):
Config.clear()
Config.define_bool("VaR", False, "turn me true")
Config.define_int("v1", 0, "var")
Config.define_int("nm.v2", 0, "var")
Config.load_conf()
self.assertEqual(Config.get_var("v1"), 1)
self.assertEqual(Config.get_var("nm.v2"), 2)
self.assertEqual(Config.get_var("VAr"), True)
def test_define_bool(self):
Config.clear()
Config.define_bool("test", 0, "int")
Config.define_bool("test2", "false", "str")
Config.define_bool("test3", "truE", "str")
self.assertDictEqual(Config.get_dict(), {
'test': False,
'test2': False,
'test3': True
})
def test_define_str(self):
Config.clear()
Config.define_str("test", 5, "A test for int")
Config.define_str("test_f", 5., "A test for int with a float var")
Config.define_str("test_f2", 5.6, "A test for int with a float var")
Config.define_str("test_str", "5.6", "A test for int with a float var")
self.assertDictEqual(Config.get_dict(), {
'test': "5",
'test_f': "5.0",
'test_f2': "5.6",
'test_str': "5.6"
})
def test_define_int(self):
Config.clear()
# int and flat float can be assign as int
Config.define_int("a", 5, "A test for int")
Config.define_int("b", 5., "A test for int with a float var")
# But not float
with self.assertRaises(ValueError):
Config.define_int("c", 5.5, "A test for int with a float var")
# Can't assign twice the same variable
with self.assertRaises(KeyError):
Config.define_int("a", 5, "A test for int")
self.assertDictEqual(Config.get_dict(), {'a': 5, 'b': 5})
def test_define_list(self):
Config.clear()
Config.define_float_list("test_float", [5., 6, "7."],
"A test for float")
Config.define_int_list("test_int_float_valid", [5., 5],
"A test for int with a float var")
Config.define_str_list("str", [], "A test for int with a float var")
self.assertDictEqual(
Config.get_dict(), {
'test_float': [5.0, 6.0, 7.0],
'test_int_float_valid': [5, 5],
"str": []
})
def test_define_float(self):
Config.clear()
Config.define_float("a", 5, "A test for float")
Config.define_float("b", 5., "A test for float with a float var")
Config.define_float("c", 5.6, "A test for float with a float var")
Config.define_float("d", "5.6", "A test for float with a str var")
with self.assertRaises(ValueError):
Config.define_float("e", "bob",
"A test for float with a unparsable str")
self.assertDictEqual(Config.get_dict(), {
'a': 5.0,
'b': 5.0,
'c': 5.6,
'd': 5.6
})
def test_load_conf_cmd(self, mock_args):
Config.clear()
Config.define_bool("VaR", False, "turn me true")
Config.define_int("v1", 1, "var")
Config.define_int("v2", 2, "var")
Config.define_float_list("list", [5., 6, "7."], "A test for list")
with Config.namespace("n1"):
Config.define_int("v1", 1, "var")
Config.define_int("v2", 2, "var")
Config.define_bool("V3", True, "turn me false")
Config.load_conf()
self.assertEqual(Config.get_var("v1"), 2)
self.assertEqual(Config.get_var("v2"), 3)
self.assertEqual(Config.get_var("n1.v1"), 2)
self.assertEqual(Config.get_var("n1.v2"), 3)
self.assertEqual(Config.get_var("n1.v3"), False)
self.assertEqual(Config.get_var("VAr"), True)
self.assertEqual(Config.get_var("list"), [1.0, 2.0, 3.0])
def test_namespace(self):
Config.clear()
Config.define_int("namespace1.test_int", 5, "A test for int")
with Config.namespace("namespace2"):
Config.define_int("test_int", 5, "A test for int")
Config.define_int("subnamespace.test_int", 5, "A test for int")
with Config.namespace("subnamespace2"):
Config.define_int("plop", 4, "test of subnamespace")
self.assertDictEqual(
Config.get_dict(), {
'namespace1': {
'test_int': 5
},
'namespace2': {
'test_int': 5,
'subnamespace': {
'test_int': 5
},
'subnamespace2': {
'plop': 4
}
}
})
def test_define_enum(self):
Config.clear()
Config.define_enum("test", "a", ["a", "b"], "a or b")
self.assertDictEqual(Config.get_dict(), {"test": "a"})