Mapytex/pymath/calculus/test/test_fraction.py

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

import unittest

from pymath.calculus.fraction import Fraction


class TestFraction(unittest.TestCase):
    """Testing functions from pymath.calculus.Fraction"""

    def setUp(self):
        self.listFrom = [Fraction(1, 3), 1]
        self.listAgainst = [Fraction(1, 3),
                            Fraction(2, 3),
                            Fraction(4, 5),
                            Fraction(-1, 3),
                            Fraction(1, -3),
                            1,
                            ]

    def test_add(self):
        ans = [
            [
                Fraction(
                    2, 3), 1, Fraction(
                    17, 15), 0, 0, Fraction(
                    4, 3)], [
                        Fraction(
                            4, 3), Fraction(
                                5, 3), Fraction(
                                    9, 5), Fraction(
                                        2, 3), Fraction(
                                            2, 3), 2]]

        for (i, f1) in enumerate(self.listFrom):
            for (j, f2) in enumerate(self.listAgainst):
                res = f1 + f2
                self.assertEqual(res, ans[i][j])

    def test_sub(self):
        ans = [[0, Fraction(-1, 3), Fraction(-7, 15), Fraction(2, 3), Fraction(2, 3), Fraction(-2, 3)],
               [Fraction(2, 3), Fraction(1, 3), Fraction(1, 5), Fraction(4, 3), Fraction(4, 3), 0]
               ]

        for (i, f1) in enumerate(self.listFrom):
            for (j, f2) in enumerate(self.listAgainst):
                res = f1 - f2
                self.assertEqual(res, ans[i][j])

    def test_neg(self):
        ans = [Fraction(-1, 3),
               Fraction(-2, 3),
               Fraction(-4, 5),
               Fraction(1, 3),
               Fraction(1, 3),
               -1
               ]
        for (j, f) in enumerate(self.listAgainst):
            res = -f
            self.assertEqual(res, ans[j])

    def test_mul(self):
        ans = [[Fraction(1, 9), Fraction(2, 9), Fraction(4, 15), Fraction(-1, 9), Fraction(-1, 9), Fraction(
            1, 3)], [Fraction(1, 3), Fraction(2, 3), Fraction(4, 5), Fraction(-1, 3), Fraction(1, -3), 1]]

        for (i, f1) in enumerate(self.listFrom):
            for (j, f2) in enumerate(self.listAgainst):
                res = f1 * f2
                self.assertEqual(res, ans[i][j])

    def test_truediv(self):
        ans = [[1, Fraction(1, 2), Fraction(5, 12), -1, -1, Fraction(1, 3)],
               [3, Fraction(3, 2), Fraction(5, 4), -3, -3, 1]
               ]

        for (i, f1) in enumerate(self.listFrom):
            for (j, f2) in enumerate(self.listAgainst):
                res = f1 / f2
                self.assertEqual(res, ans[i][j])

    def test_lt(self):
        pass

    def test_le(self):
        pass

    def test_tex(self):
        f = Fraction(2, 3)
        ans = "\\frac{ 2 }{ 3 }"
        self.assertEqual(f.__tex__(), ans)

    def test_txt(self):
        f = Fraction(2, 3)
        ans = "2 / 3"
        self.assertEqual(f.__txt__(), ans)

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

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