#!/usr/bin/env python
# encoding: utf-8


import unittest

from pymath.render import tex, txt,p2i
from pymath.fraction import Fraction
from pymath.polynom import Polynom
from pymath.operator import op



class TestTexRender(unittest.TestCase):
    """Testing functions from pymath.renders.tex"""

    def test_type_render_int(self):
        self.assertEqual(tex([2]), "2")

    def test_type_render_str(self):
        self.assertEqual(tex(["a"]), "a")

    def test_type_render_fraction(self):
        self.assertEqual(tex([Fraction(1,2)]), "\\frac{ 1 }{ 2 }")

    def test_type_render_poly(self):
        P = Polynom([1,2,3])
        self.assertEqual(tex([P]), "3 x^{  2 } + 2 x + 1")

    def test_add_interger(self):
        exps = [ [2, 3, op.add ],
                [2, -3, op.add ],
                [-2, 3, op.add ],
                ]
        wanted_render = [ "2 + 3",
                "2 + ( -3 )",
                "-2 + 3",
                ]
        for (i,e) in enumerate(exps):
            rend = tex(e)
            self.assertEqual(rend, wanted_render[i])

    def test_add_letter(self):
        exps = [[2, "a",  op.add ],
                ["a", 3,  op.add ],
                [-2, "a", op.add ],
                ["a", -2, op.add ],
                ]
        wanted_render = [ "2 + a",
                "a + 3",
                "-2 + a",
                "a + ( -2 )",
                ]
        for (i,e) in enumerate(exps):
            rend = tex(e)
            self.assertEqual(rend, wanted_render[i])

    def test_add_fraction(self):
        exps = [[2, Fraction(1,2), op.add],
                [Fraction(1,2), 3, op.add],
                ]
        wanted_render = [ "2 + \\frac{ 1 }{ 2 }",
                "\\frac{ 1 }{ 2 } + 3",
                ]
        for (i,e) in enumerate(exps):
            rend = tex(e)
            self.assertEqual(rend, wanted_render[i])

    def test_add_poly(self):
        exps = [[2, Polynom([1,2,3]), op.mul],
                [Polynom([1,2,3]), 2, op.mul],
                [Polynom([1,2,3]), Polynom([4,5,6]), op.mul],
                ]
        wanted_render = [ "2 ( 3 x^{  2 } + 2 x + 1 )",
                "( 3 x^{  2 } + 2 x + 1 ) \\times 2",
                "( 3 x^{  2 } + 2 x + 1 ) ( 6 x^{  2 } + 5 x + 4 )",
                ]
        for (i,e) in enumerate(exps):
            rend = tex(e)
            self.assertEqual(rend, wanted_render[i])

    def test_mult_interger(self):
        exps = [[2, 3,  op.mul],
                [2, -3, op.mul],
                [-2, 3, op.mul],
                ]
        wanted_render = [ "2 \\times 3",
                "2 \\times ( -3 )",
                "-2 \\times 3",
                ]
        for (i,e) in enumerate(exps):
            rend = tex(e)
            self.assertEqual(rend, wanted_render[i])

    def test_mult_letter(self):
        exps = [[2, "a",  op.mul],
                ["a", 3,  op.mul],
                [-2, "a", op.mul],
                ["a", -2, op.mul],
                ]
        wanted_render = [ "2 a", "a \\times 3", "-2 a", "a \\times ( -2 )"]
        for (i,e) in enumerate(exps):
            rend = tex(e)
            self.assertEqual(rend, wanted_render[i])

    def test_mult_fraction(self):
        exps = [ [2, Fraction(1,2), op.mul], [Fraction(1,2), 3, op.mul]]
        wanted_render = [ "2 \\times \\frac{ 1 }{ 2 }", "\\frac{ 1 }{ 2 } \\times 3"]
        for (i,e) in enumerate(exps):
            rend = tex(e)
            self.assertEqual(rend, wanted_render[i])

    def test_mult_poly(self):
        exps = [[2, Polynom([1,2,3]), op.mul],
                [Polynom([1,2,3]), 2, op.mul],
                [Polynom([1,2,3]), Polynom([4,5,6]), op.mul],
                ]
        wanted_render = [ "2 ( 3 x^{  2 } + 2 x + 1 )",
                "( 3 x^{  2 } + 2 x + 1 ) \\times 2",
                "( 3 x^{  2 } + 2 x + 1 ) ( 6 x^{  2 } + 5 x + 4 )",
                ]
        for (i,e) in enumerate(exps):
            rend = tex(e)
            self.assertEqual(rend, wanted_render[i])

    def test_parentheses_int(self):
        exps = [\
            [ 2, 3, op.add, 4, op.mul],\
            [ 2, 3, op.mul, 4, op.add],\
            [ 2, 3, 4, op.mul, op.add],\
            [ 2, 3, 4, op.add, op.add],\
            [ 2, 3, 4, op.add, op.sub],\
            ]
        wanted_render = [\
                '( 2 + 3 ) \\times 4',\
                '2 \\times 3 + 4',\
                '2 + 3 \\times 4',\
                '2 + 3 + 4',\
                '2 - ( 3 + 4 )',\
                ]
        for (i,e) in enumerate(exps):
            rend = tex(e)
            self.assertEqual(rend, wanted_render[i])

    def test_parentheses_poly(self):
        P = Polynom([1,2,3])
        Q = Polynom([4,5,6])
        exps = [\
            [ 2, P, op.add],\
            [ 2, P, op.sub],\
            [ 2, P, P, op.mul, op.sub],\
            [ Q, P, op.add],\
            [ Q, P, op.sub],\
            ]
        wanted_render = [\
                '2 + 3 x^{  2 } + 2 x + 1' ,\
                '2 - ( 3 x^{  2 } + 2 x + 1 )' ,\
                '2 - ( 3 x^{  2 } + 2 x + 1 ) ( 3 x^{  2 } + 2 x + 1 )' ,\
                '6 x^{  2 } + 5 x + 4 + 3 x^{  2 } + 2 x + 1' ,\
                '6 x^{  2 } + 5 x + 4 - ( 3 x^{  2 } + 2 x + 1 )' ,\
                ]
        for (i,e) in enumerate(exps):
            rend = tex(e)
            self.assertEqual(rend, wanted_render[i])

    def test_slash(self):
        pass




class TesttxtRender(unittest.TestCase):
    """Testing functions from pymath.renders.txt"""

    def test_type_render_int(self):
        self.assertEqual(txt([2]), "2")

    def test_type_render_str(self):
        self.assertEqual(txt(["a"]), "a")

    def test_type_render_fraction(self):
        self.assertEqual(txt([Fraction(1,2)]), "1 / 2")

    def test_mult_interger(self):
        exps = [ [2, 3, op.mul], \
                [2, -3, op.mul], \
                [-2, 3, op.mul]]
        wanted_render = [ "2 * 3", "2 * ( -3 )", "-2 * 3"]
        for (i,e) in enumerate(exps):
            rend = txt(e)
            self.assertEqual(rend, wanted_render[i])

    def test_mult_letter(self):
        exps = [ [2, "a", op.mul], \
                ["a", 3, op.mul], \
                [-2, "a", op.mul], \
                ["a", -2, op.mul],
                ["a", -2, op.mul, -2, op.mul],
                ["a", -2, op.mul, "a", op.mul],
                ]
        wanted_render = [ "2 a",
                "a * 3",
                "-2 a",
                "a * ( -2 )",
                "a * ( -2 ) * ( -2 )",
                "a * ( -2 ) a",
                ]
        for (i,e) in enumerate(exps):
            rend = txt(e)
            self.assertEqual(rend, wanted_render[i])

    def test_mult_fraction(self):
        exps = [ [2, Fraction(1,2), op.mul], \
                [Fraction(1,2), 3,  op.mul]]
        wanted_render = [ "2 * 1 / 2", "1 / 2 * 3"]
        for (i,e) in enumerate(exps):
            rend = txt(e)
            self.assertEqual(rend, wanted_render[i])

    def test_parentheses(self):
        mul = op.get_op("*", 2)
        add = op.get_op("+", 2)
        exps = [\
            [ 2, 3, add, 4, mul],\
            [ 2, 3, mul, 4, add],\
            [ 2, 3, 4, mul, add],\
            [ 2, 3, 4, add, add],\
            ]
        wanted_render = [\
                '( 2 + 3 ) * 4',\
                '2 * 3 + 4',\
                '2 + 3 * 4',\
                '2 + 3 + 4',\
                ]
        for (i,e) in enumerate(exps):
            rend = txt(e)
            self.assertEqual(rend, wanted_render[i])

    def test_slash(self):
        pass


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








# -----------------------------
# Reglages pour 'vim'
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
# cursor: 16 del