# This file is part of TRS (http://math.kompiler.org)
#
# TRS is free software: you can redistribute it and/or modify it under the
# terms of the GNU Affero General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option) any
# later version.
#
# TRS is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE.  See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License
# along with TRS.  If not, see <http://www.gnu.org/licenses/>.
from tests.rulestestcase import RulesTestCase as TestCase, rewrite


class TestLeidenOefenopgave(TestCase):
    def test_1_1(self):
        self.assertRewrite([
            '-5(x ^ 2 - 3x + 6)',
            '-(5x ^ 2 + 5 * -3x + 5 * 6)',
            '-(5x ^ 2 + (-15)x + 5 * 6)',
            '-(5x ^ 2 + (-15)x + 30)',
            '-(5x ^ 2 - 15x + 30)',
            '-5x ^ 2 - -15x - 30',
            '-5x ^ 2 + 15x - 30',
        ])

        #for exp, solution in [
        #        ('-5(x^2 - 3x + 6)',     '-30 + 15x - 5x ^ 2'),
        #        ('(x+1)^2',              'x ^ 2 + 2x + 1'),
        #        ('(x-1)^2',              'x ^ 2 - 2x + 1'),
        #        ('(2x+x)*x',             '3x ^ 2'),
        #        ('-2(6x-4)^2*x',         '-72x ^ 3 + 96x ^ 2 + 32x'),
        #        ('(4x + 5) * -(5 - 4x)', '16x^2 - 25'),
        #        ]:
        #    self.assertEqual(str(rewrite(exp)), solution)

    def test_1_2(self):
        self.assertRewrite([
            '(x + 1) ^ 3',
            '(x + 1)(x + 1) ^ 2',
            '(x + 1)(x + 1)(x + 1)',
            '(x * x + x * 1 + 1x + 1 * 1)(x + 1)',
            '(x * x + x + 1x + 1 * 1)(x + 1)',
            '(x * x + x + x + 1 * 1)(x + 1)',
            '(x * x + x + x + 1)(x + 1)',
            '(x * x + (1 + 1)x + 1)(x + 1)',
            '(x * x + 2x + 1)(x + 1)',
            '(x ^ (1 + 1) + 2x + 1)(x + 1)',
            '(x ^ 2 + 2x + 1)(x + 1)',
            'x ^ 2 * x + x ^ 2 * 1 + 2x * x + 2x * 1 + 1x + 1 * 1',
            'x ^ 2 * x + x ^ 2 + 2x * x + 2x * 1 + 1x + 1 * 1',
            'x ^ 2 * x + x ^ 2 + 2x * x + 2x + 1x + 1 * 1',
            'x ^ 2 * x + x ^ 2 + 2x * x + 2x + x + 1 * 1',
            'x ^ 2 * x + x ^ 2 + 2x * x + 2x + x + 1',
            'x ^ 2 * x + x ^ 2 + 2x * x + (2 + 1)x + 1',
            'x ^ 2 * x + x ^ 2 + 2x * x + 3x + 1',
            'x ^ (2 + 1) + x ^ 2 + 2x * x + 3x + 1',
            'x ^ 3 + x ^ 2 + 2x * x + 3x + 1',
            'x ^ 3 + x ^ 2 + 2x ^ (1 + 1) + 3x + 1',
            'x ^ 3 + x ^ 2 + 2x ^ 2 + 3x + 1',
            'x ^ 3 + (1 + 2)x ^ 2 + 3x + 1',
            'x ^ 3 + 3x ^ 2 + 3x + 1',
        ])

    def test_1_3(self):
        self.assertRewrite([
            '(x + 1) ^ 2',
            '(x + 1)(x + 1)',
            'x * x + x * 1 + 1x + 1 * 1',
            'x * x + x + 1x + 1 * 1',
            'x * x + x + x + 1 * 1',
            'x * x + x + x + 1',
            'x * x + (1 + 1)x + 1',
            'x * x + 2x + 1',
            'x ^ (1 + 1) + 2x + 1',
            'x ^ 2 + 2x + 1',
        ])

    def test_1_4(self):
        self.assertRewrite([
            '(x - 1) ^ 2',
            '(x - 1)(x - 1)',
            'x * x + x * -1 + (-1)x + (-1) * -1',
            'x * x + x * -1 + (-1)x - -1',
            'x * x + x * -1 + (-1)x + 1',
            'x * x - x * 1 + (-1)x + 1',
            'x * x - x + (-1)x + 1',
            'x * x - x - 1x + 1',
            'x * x - x - x + 1',
            'x * x + (1 + 1) * -x + 1',
            'x * x + 2 * -x + 1',
            'x * x - 2x + 1',
            'x ^ (1 + 1) - 2x + 1',
            'x ^ 2 - 2x + 1',
        ])

    def test_1_4_1(self):
        self.assertRewrite([
            'x * -1 + 1x',
            'x * -1 + x',
            '-x * 1 + x',
            '-x + x',
            '(-1 + 1)x',
            '0x',
            '0',
        ])

    def test_1_4_2(self):
        self.assertRewrite([
            'x * -1 - 1x',
            'x * -1 - x',
            '-x * 1 - x',
            '-x - x',
            '(1 + 1) * -x',
            '2 * -x',
            '-2x',
        ])

    def test_1_4_3(self):
        self.assertRewrite([
            'x * -1 + x * -1',
            '-x * 1 + x * -1',
            '-x + x * -1',
            '-x - x * 1',
            '-x - x',
            '(1 + 1) * -x',
            '2 * -x',
            '-2x',
        ])

    def test_1_5(self):
        self.assertRewrite([
            '(2x + x)x',
            '(2 + 1)x * x',
            '3x * x',
            '3x ^ (1 + 1)',
            '3x ^ 2',
        ])

    def test_1_7(self):
        self.assertRewrite([
            '(4x + 5) * -(5 - 4x)',
            '-(4x + 5)(5 - 4x)',
            '-(4x * 5 + 4x * -4x + 5 * 5 + 5 * -4x)',
            '-(20x + 4x * -4x + 5 * 5 + 5 * -4x)',
            '-(20x + (-16)x * x + 5 * 5 + 5 * -4x)',
            '-(20x + (-16)x * x + 25 + 5 * -4x)',
            '-(20x + (-16)x * x + 25 + (-20)x)',
            '-(20x - 16x * x + 25 + (-20)x)',
            '-(20x - 16x * x + 25 - 20x)',
            '-((1 - 1)20x - 16x * x + 25)',
            '-(0 * 20x - 16x * x + 25)',
            '-(0 - 16x * x + 25)',
            '-(-16x * x + 25)',
            '-(-16x ^ (1 + 1) + 25)',
            '-(-16x ^ 2 + 25)',
            '--16x ^ 2 - 25',
            '16x ^ 2 - 25',
        ])

    def test_2(self):
        pass

    def test_3(self):
        pass

    def test_4_1(self):
        self.assertRewrite([
            '2/15 + 1/4',
            '8 / 60 + 15 / 60',
            '(8 + 15) / 60',
            '23 / 60',
        ])

    def test_4_2(self):
        self.assertRewrite([
            '2 / 7 - 4 / 11',
            '22 / 77 - 28 / 77',
            '(22 - 28) / 77',
            '(-6) / 77',
            '-6 / 77',
        ])

    def test_4_3(self):
        self.assertRewrite([
            '(7/3)(3/5)',
            '(7 * 3) / (3 * 5)',
            '7 / 5',
        ])

    def test_4_4(self):
        self.assertRewrite([
            '(3/4) / (5/6)',
            '3 / (4 * 5 / 6)',
            '3 / ((4 * 5) / 6)',
            '3 / (20 / 6)',
            '(3 * 6) / 20',
            '18 / 20',
            '9 / 10'])

    def test_4_5(self):
        self.assertRewrite([
            '1/4 * 1/x',
            '(1 * 1) / (4x)',
            '1 / (4x)',
        ])

    def test_4_6(self):
        self.assertRewrite([
            '(3 / x^2) / (x / 7)',
            '3 / x ^ 2 / (1 / 7 * x)',
            '3 / (x ^ 2 * 1 / 7x)',
            '3 / (x ^ (2 + 1)1 / 7)',
            '3 / (x ^ 3 * 1 / 7)',
            '(7 * 3) / x ^ 3',
            '21 / x ^ 3',
        ])

    def test_4_7(self):
        self.assertEvaluates('1 / x + 2 / (x + 1)', '(3x + 1) / (x(x + 1))')