Add getall/getrawall for nested params, ROOT_KEY constant, tests and docs
This commit is contained in:
@@ -98,6 +98,45 @@ def test_config_validator():
|
||||
assert config.get("level", cat="test") == "DEBUG"
|
||||
|
||||
|
||||
def test_config_getraw_setraw():
|
||||
config.reset()
|
||||
config.register(name="path", val="val1", cat="model1")
|
||||
config.register(name="opt.temperature", val="val2", cat="model1")
|
||||
config.register(name="path", val="val3", cat="global")
|
||||
|
||||
assert config.getraw("model1.path") == "val1"
|
||||
assert config.getraw("model1.opt.temperature") == "val2"
|
||||
assert config.getraw("global.path") == "val3"
|
||||
|
||||
config.setraw("model1.path", "new_val")
|
||||
assert config.getraw("model1.path") == "new_val"
|
||||
|
||||
|
||||
def test_config_getall_getrawall():
|
||||
config.reset()
|
||||
config.register(name="coder", val="llama-coder", cat="models")
|
||||
config.register(name="coder.thinking", val="full", cat="models")
|
||||
config.register(name="coder.temperature", val="0.3", cat="models")
|
||||
config.register(name="chatter", val="gemmini", cat="models")
|
||||
config.register(name="chatter.thinking", val="none", cat="models")
|
||||
config.register(name="chatter.temperature", val="0.9", cat="models")
|
||||
|
||||
assert config.get("coder", cat="models") == "llama-coder"
|
||||
|
||||
all_coder = config.getall("coder", cat="models")
|
||||
assert all_coder == {
|
||||
"$root$": "llama-coder",
|
||||
"thinking": "full",
|
||||
"temperature": "0.3",
|
||||
}
|
||||
|
||||
rawall = config.getrawall("models")
|
||||
assert rawall == {
|
||||
"coder": {"$root$": "llama-coder", "thinking": "full", "temperature": "0.3"},
|
||||
"chatter": {"$root$": "gemmini", "thinking": "none", "temperature": "0.9"},
|
||||
}
|
||||
|
||||
|
||||
def run_tests():
|
||||
tests = [
|
||||
test_config_manager_register_and_get,
|
||||
@@ -108,6 +147,8 @@ def run_tests():
|
||||
test_config_parameter_get_parameter,
|
||||
test_config_set_value,
|
||||
test_config_validator,
|
||||
test_config_getraw_setraw,
|
||||
test_config_getall_getrawall,
|
||||
]
|
||||
|
||||
passed = 0
|
||||
|
||||
Reference in New Issue
Block a user