"""CLI chart workflows."""
from __future__ import annotations
import pytest
from pytest_aitest import Agent, Provider
from conftest import assert_cli_exit_codes, assert_regex, unique_path, DEFAULT_RETRIES, DEFAULT_TIMEOUT_MS
pytestmark = [pytest.mark.aitest, pytest.mark.cli]
@pytest.mark.asyncio
async def test_cli_chart_workflows(aitest_run, excel_cli_server, excel_cli_skill):
agent = Agent(
name="cli-chart-workflows",
provider=Provider(model="azure/gpt-4.1", rpm=10, tpm=10000),
cli_servers=[excel_cli_server],
skill=excel_cli_skill,
max_turns=25,
retries=DEFAULT_RETRIES,
)
messages = None
prompt = f"""
Create a sales analysis chart:
1. Create a new Excel file at {unique_path('chart-from-table-cli')}
2. Enter this sales data in A1:C5:
Product, Q1 Sales, Q2 Sales
Laptop, 45000, 52000
Phone, 38000, 41000
Tablet, 22000, 28000
Monitor, 15000, 18000
3. Convert the data to a table called "SalesData"
4. Create a column chart from the table
5. Position it below the data
6. Save and close
"""
result = await aitest_run(agent, prompt, messages=messages, timeout_ms=DEFAULT_TIMEOUT_MS)
assert result.success
assert_cli_exit_codes(result)
assert_regex(result.final_response, r"(?i)(salesdata|chart|created)")
messages = result.messages
prompt = f"""
I need a chart that doesn't overlap my data:
1. Create a new Excel file at {unique_path('chart-position-cli')}
2. Put this budget data in A1:C6:
Month, Revenue, Expenses
January, 50000, 35000
February, 55000, 38000
March, 48000, 32000
April, 62000, 41000
May, 58000, 39000
3. Create a line chart from the data
4. Make sure the chart is positioned BELOW row 6 so it doesn't cover the data
5. Read chart info and confirm the position
6. Save and close
"""
result = await aitest_run(agent, prompt, messages=messages, timeout_ms=DEFAULT_TIMEOUT_MS)
assert result.success
assert_cli_exit_codes(result)
assert_regex(result.final_response, r"(?i)(chart|row [7-9]|\$[A-Z]+\$[7-9])")
messages = result.messages
prompt = f"""
Create a dashboard with multiple chart types:
1. Create a new Excel file at {unique_path('multi-chart-cli')}
2. Enter market data in A1:C5:
Company, Revenue, Market Share
Alpha, 500000, 35
Beta, 400000, 28
Gamma, 300000, 22
Delta, 200000, 15
3. Convert to a table called "MarketData"
4. Create a PIE chart showing Market Share
5. Create a BAR chart showing Revenue
6. List charts and confirm both exist without overlapping
7. Save and close
"""
result = await aitest_run(agent, prompt, messages=messages, timeout_ms=DEFAULT_TIMEOUT_MS)
assert result.success
assert_cli_exit_codes(result)
assert_regex(result.final_response, r"(?i)(pie|bar|2 chart|two chart)")
messages = result.messages
prompt = f"""
Create a chart with precise cell-based positioning:
1. Create a new Excel file at {unique_path('chart-targetrange-cli')}
2. Enter quarterly data in A1:E5:
Region, Q1, Q2, Q3, Q4
North, 1000, 1200, 1100, 1400
South, 800, 900, 950, 1000
East, 1500, 1600, 1450, 1700
West, 600, 700, 650, 800
3. Create a bar chart positioned to the right of the data
4. Verify the chart was created and its position
5. Save and close
"""
result = await aitest_run(agent, prompt, messages=messages, timeout_ms=DEFAULT_TIMEOUT_MS)
assert result.success
assert_cli_exit_codes(result)
assert_regex(result.final_response, r"(?i)(chart|created)")
