24 lines
713 B
Python
24 lines
713 B
Python
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()
|