# -*- coding: utf-8 -*-
import stl
from nose2.tools import params
import unittest
from sympy import Symbol

ex1 = ('x1 > 2', stl.LinEq(
    (stl.Var(1, Symbol("x1"), stl.ast.t_sym),),
    ">",
    2.0
))
ex1_ = ('x1 > a?', stl.LinEq(
    (stl.Var(1, Symbol("x1"), stl.ast.t_sym),),
    ">",
    Symbol("a?")
))

ex1__ = ('x1', stl.AtomicPred('x1'))

i1 = stl.Interval(0., 1.)
i1_ = stl.Interval(0., Symbol("b?"))
i2 = stl.Interval(2., 3.)
ex2 = ('◇[0,1](x1 > 2)', stl.F(i1, ex1[1]))
ex3 = ('□[2,3]◇[0,1](x1 > 2)', stl.G(i2, ex2[1]))
ex4 = ('(x1 > 2) or ((x1 > 2) or (x1 > 2))', 
       stl.Or((ex1[1], ex1[1], ex1[1])))
ex5 = ('G[0, b?](x1 > a?)',
       stl.G(i1_, ex1_[1]))
ex6 = ('◇[0,1](x1)', stl.F(i1, ex1__[1]))
 
class TestSTLParser(unittest.TestCase):
    @params(ex1, ex2, ex3, ex4, ex5, ex6)
    def test_stl(self, phi_str, phi):
        self.assertEqual(stl.parse(phi_str), phi)