equicompiler-algebraic-port.../tests/test_version_override.py

21 lines
539 B
Python

import unittest
from equicompiler_algebraic_portfolio_dsl_to_.core import parse_dsl_to_ir
class TestVersionOverride(unittest.TestCase):
def test_version_override_via_dsl(self):
dsl = (
"version: 0.2\n"
"assets: AAPL, MSFT\n"
"objectives: maximize_return\n"
"constraints: max_drawdown=0.2, var=0.95"
)
ir = parse_dsl_to_ir(dsl)
self.assertIn("version", ir)
self.assertEqual(ir["version"], "0.2")
if __name__ == "__main__":
unittest.main()