import unittest

from equicompiler_algebraic_portfolio_dsl_to_.core import parse_dsl_to_ir


class TestExecutionPolicy(unittest.TestCase):
    def test_parse_execution_policy(self):
        dsl = (
            "assets: AAPL, MSFT\n"
            "objectives: maximize_return\n"
            "constraints: max_drawdown=0.2, var=0.95\n"
            "execution_policy: immediate, time_slice=1ms"
        )
        ir = parse_dsl_to_ir(dsl)
        self.assertIn("execution_policy", ir)
        policies = ir["execution_policy"]
        self.assertIsInstance(policies, list)
        self.assertIn("immediate", policies)
        self.assertIn("time_slice=1ms", policies)


if __name__ == "__main__":
    unittest.main()