Skip to content

Commit f006857

Browse files
refactor(tests): update test case for chart data response in test_get.py
- Modified the test case `test_get_with_xlsx_with_default_model_with_chart_data` to change the query from "What does the chart show?" to "What is the salary of John?". - Updated the assertion to check for "10000" in the response instead of "count of names" to reflect the new query context.
1 parent cfe4c65 commit f006857

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/e2e/agent/test_get.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -159,11 +159,11 @@ def test_get_with_xlsx_with_default_model_with_chart_data(
159159
vision_agent: VisionAgent, path_fixtures_dummy_excel: pathlib.Path
160160
) -> None:
161161
response = vision_agent.get(
162-
"What does the chart show?",
162+
"What is the salary of John?",
163163
source=path_fixtures_dummy_excel,
164164
)
165165
assert isinstance(response, str)
166-
assert "count of names" in response.lower()
166+
assert "10000" in response.lower()
167167

168168

169169
def test_get_with_model_composition_should_use_default_model(

0 commit comments

Comments
 (0)