From 5e13e3add49f73166724658fa59589fda24ff78c Mon Sep 17 00:00:00 2001 From: Monte Hoover Date: Mon, 20 Jul 2026 14:17:01 -0400 Subject: [PATCH] Update generated tests to get python parameter names instead of raw API-provider parameter names. (#7597) --- cli/openbb_cli/codegen/test_gen.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/cli/openbb_cli/codegen/test_gen.py b/cli/openbb_cli/codegen/test_gen.py index 4028e723976..6784a76026a 100644 --- a/cli/openbb_cli/codegen/test_gen.py +++ b/cli/openbb_cli/codegen/test_gen.py @@ -5,6 +5,8 @@ from __future__ import annotations from dataclasses import dataclass from typing import TYPE_CHECKING, Any +from openbb_cli.codegen.pydantic_gen import safe_field_name + if TYPE_CHECKING: from openbb_cli.codegen.fetcher_gen import GeneratedFetcher @@ -251,13 +253,14 @@ def _derive_test_params( param_providers = raw.get("providers") or [] if param_providers and provider_name not in param_providers: continue + field_name, _ = safe_field_name(name) if raw.get("required"): value = _spec_supplied_value(raw) if value is None: return None - out[name] = value + out[field_name] = value elif _is_date_param(name): - out[name] = _refresh_date_shape(name, "1970-01-01") + out[field_name] = _refresh_date_shape(name, "1970-01-01") return out