apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-04-25 00:36:34 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fix black formatting for Python test files

Browse source
This commit is contained in:
Adil Hafeez 2026-03-15 08:49:22 +00:00
parent c4634d0034
commit 74f9b44378
2 changed files with 103 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

138
cli/test/test_config_generator.py
View file

@ -11,9 +11,11 @@ def cleanup_env(monkeypatch):
monkeypatch.undo() monkeypatch.undo()
@pytest.mark.parametrize("plano_config", [ @pytest.mark.parametrize(
# Case 1: LLM provider config "plano_config",
""" [
# Case 1: LLM provider config
"""
version: v0.1.0 version: v0.1.0
listeners: listeners:
@ -44,8 +46,8 @@ llm_providers:
tracing: tracing:
random_sampling: 100 random_sampling: 100
""", """,
# Case 2: Agent config # Case 2: Agent config
""" """
version: v0.3.0 version: v0.3.0
agents: agents:
@ -86,7 +88,9 @@ model_providers:
- access_key: ${OPENAI_API_KEY} - access_key: ${OPENAI_API_KEY}
model: openai/gpt-4o model: openai/gpt-4o
""", """,
], ids=["llm_provider_config", "agent_config"]) ],
ids=["llm_provider_config", "agent_config"],
)
def test_validate_and_render_happy_path(monkeypatch, plano_config): def test_validate_and_render_happy_path(monkeypatch, plano_config):
monkeypatch.setenv("PLANO_CONFIG_FILE", "fake_plano_config.yaml") monkeypatch.setenv("PLANO_CONFIG_FILE", "fake_plano_config.yaml")
monkeypatch.setenv("PLANO_CONFIG_SCHEMA_FILE", "fake_plano_config_schema.yaml") monkeypatch.setenv("PLANO_CONFIG_SCHEMA_FILE", "fake_plano_config_schema.yaml")
@ -310,49 +314,95 @@ def test_validate_and_render_schema_tests(monkeypatch, plano_config_test_case):
validate_and_render_schema() validate_and_render_schema()
@pytest.mark.parametrize("listeners,expected_providers,expected_llm_gateway,expected_prompt_gateway", [ @pytest.mark.parametrize(
# Case 1: With prompt gateway (ingress + egress) "listeners,expected_providers,expected_llm_gateway,expected_prompt_gateway",
( [
{ # Case 1: With prompt gateway (ingress + egress)
"ingress_traffic": {"address": "0.0.0.0", "port": 10000, "timeout": "30s"}, (
"egress_traffic": {"address": "0.0.0.0", "port": 12000, "timeout": "30s"},
},
[
{ {
"name": "egress_traffic", "type": "model_listener", "port": 12000, "ingress_traffic": {
"address": "0.0.0.0", "timeout": "30s", "address": "0.0.0.0",
"model_providers": [{"model": "openai/gpt-4o", "access_key": "test_key"}], "port": 10000,
"timeout": "30s",
},
"egress_traffic": {
"address": "0.0.0.0",
"port": 12000,
"timeout": "30s",
},
}, },
{ [
"name": "ingress_traffic", "type": "prompt_listener", "port": 10000, {
"address": "0.0.0.0", "timeout": "30s", "name": "egress_traffic",
}, "type": "model_listener",
], "port": 12000,
{ "address": "0.0.0.0",
"address": "0.0.0.0", "model_providers": [{"access_key": "test_key", "model": "openai/gpt-4o"}], "timeout": "30s",
"name": "egress_traffic", "type": "model_listener", "port": 12000, "timeout": "30s", "model_providers": [
}, {"model": "openai/gpt-4o", "access_key": "test_key"}
{"address": "0.0.0.0", "name": "ingress_traffic", "port": 10000, "timeout": "30s", "type": "prompt_listener"}, ],
), },
# Case 2: Without prompt gateway (egress only) {
( "name": "ingress_traffic",
{"egress_traffic": {"address": "0.0.0.0", "port": 12000, "timeout": "30s"}}, "type": "prompt_listener",
[ "port": 10000,
"address": "0.0.0.0",
"timeout": "30s",
},
],
{ {
"address": "0.0.0.0", "address": "0.0.0.0",
"model_providers": [{"access_key": "test_key", "model": "openai/gpt-4o"}], "model_providers": [
"name": "egress_traffic", "port": 12000, "timeout": "30s", "type": "model_listener", {"access_key": "test_key", "model": "openai/gpt-4o"}
} ],
], "name": "egress_traffic",
{ "type": "model_listener",
"address": "0.0.0.0", "model_providers": [{"access_key": "test_key", "model": "openai/gpt-4o"}], "port": 12000,
"name": "egress_traffic", "type": "model_listener", "port": 12000, "timeout": "30s", "timeout": "30s",
}, },
None, {
), "address": "0.0.0.0",
], ids=["with_prompt_gateway", "without_prompt_gateway"]) "name": "ingress_traffic",
def test_convert_legacy_llm_providers(listeners, expected_providers, expected_llm_gateway, expected_prompt_gateway): "port": 10000,
"timeout": "30s",
"type": "prompt_listener",
},
),
# Case 2: Without prompt gateway (egress only)
(
{"egress_traffic": {"address": "0.0.0.0", "port": 12000, "timeout": "30s"}},
[
{
"address": "0.0.0.0",
"model_providers": [
{"access_key": "test_key", "model": "openai/gpt-4o"}
],
"name": "egress_traffic",
"port": 12000,
"timeout": "30s",
"type": "model_listener",
}
],
{
"address": "0.0.0.0",
"model_providers": [
{"access_key": "test_key", "model": "openai/gpt-4o"}
],
"name": "egress_traffic",
"type": "model_listener",
"port": 12000,
"timeout": "30s",
},
None,
),
],
ids=["with_prompt_gateway", "without_prompt_gateway"],
)
def test_convert_legacy_llm_providers(
listeners, expected_providers, expected_llm_gateway, expected_prompt_gateway
):
from planoai.utils import convert_legacy_listeners from planoai.utils import convert_legacy_listeners
llm_providers = [{"model": "openai/gpt-4o", "access_key": "test_key"}] llm_providers = [{"model": "openai/gpt-4o", "access_key": "test_key"}]
updated_providers, llm_gateway, prompt_gateway = convert_legacy_listeners( updated_providers, llm_gateway, prompt_gateway = convert_legacy_listeners(
listeners, llm_providers listeners, llm_providers

15
cli/test/test_version_check.py
View file

@ -52,11 +52,15 @@ class TestCheckVersionStatus:
assert status["is_outdated"] is False assert status["is_outdated"] is False
assert status["message"] is None assert status["message"] is None
@pytest.mark.parametrize("current,latest", [ @pytest.mark.parametrize(
("0.4.1", "1.0.0"), # major "current,latest",
("0.4.1", "0.5.0"), # minor [
("0.4.1", "0.4.2"), # patch ("0.4.1", "1.0.0"), # major
], ids=["major", "minor", "patch"]) ("0.4.1", "0.5.0"), # minor
("0.4.1", "0.4.2"), # patch
],
ids=["major", "minor", "patch"],
)
def test_version_outdated(self, current, latest): def test_version_outdated(self, current, latest):
status = check_version_status(current, latest) status = check_version_status(current, latest)
assert status["is_outdated"] is True assert status["is_outdated"] is True
@ -150,4 +154,3 @@ class TestVersionCheckIntegration:
status = check_version_status(current_version, latest) status = check_version_status(current_version, latest)
assert status["is_outdated"] is False assert status["is_outdated"] is False