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'))

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

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