trs/tests/test_validation.py

from unittest import TestCase
from src.validation import validate


class TestValidation(TestCase):

    def test_simple_success(self):
        self.assertTrue(validate('3a + a', '4a'))

    def test_simple_failure(self):
        self.assertFalse(validate('3a + a', '4a + 1'))

    def test_intermediate_success(self):
        self.assertTrue(validate('3a + a + b + 2b', '4a + 3b'))
        self.assertTrue(validate('a / b / (c / d)', '(ad) / (bc)'))

    def test_intermediate_failure(self):
        self.assertFalse(validate('3a + a + b + 2b', '4a + 4b'))

    #def test_advanced_failure(self):
    #    self.assertFalse(validate('(x-1)^3+(x-1)^3', '4a+4b'))