import unittest

from node import Node, Leaf
from line import generate_line


class TestLine(unittest.TestCase):

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def test_simple(self):
        l0, l1 = Leaf(1), Leaf(2)
        plus = Node('+', l0, l1)
        assert generate_line(plus, Node) == '1 + 2'

    def test_parentheses(self):
        l0, l1 = Leaf(1), Leaf(2)
        plus = Node('+', l0, l1)
        times = Node('*', plus, plus)
        assert generate_line(times, Node) == '(1 + 2) * (1 + 2)'

    def test_function(self):
        exp = Leaf('x')
        inf = Leaf('oo')
        minus_inf = Node('-', inf)
        integral = Node('int', exp, minus_inf, inf)
        assert generate_line(integral, Node) == 'int(x, -oo, oo)'

    def test_mod(self):
        l0, l1 = Leaf(1), Leaf(2)
        mod = Node('mod', l1, l0)
        assert generate_line(mod, Node) == '2 mod 1'