"""
Tests for SimulationAgent.
"""

import pytest
from dcos.agents.simulation import SimulationAgent


class TestSimulationAgent:
    def test_simulate(self):
        agent = SimulationAgent()
        result = agent.simulate("market_scenario", iterations=100)
        assert result["scenario"] == "market_scenario"
        assert "success_probability" in result

    def test_act(self):
        agent = SimulationAgent()
        result = agent.act({"scenario": "test"})
        assert "result" in result
        assert result["iterations"] == 1000