| | from sympy import simplify, srepr, Add, Mul, Pow, Rational, pi, sqrt, Symbol |
| | from latex2sympy.latex2sympy2 import latex2sympy as process_sympy |
| | import sys |
| | import os |
| | sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) |
| |
|
| | x = Symbol('x', real=True) |
| | y = Symbol('y', real=True) |
| |
|
| | |
| |
|
| |
|
| | def _Add(a, b): |
| | return Add(a, b, evaluate=False) |
| |
|
| |
|
| | def _Mul(a, b): |
| | return Mul(a, b, evaluate=False) |
| |
|
| |
|
| | def _Pow(a, b): |
| | return Pow(a, b, evaluate=False) |
| |
|
| |
|
| | def get_simple_examples(func): |
| | ''' |
| | Returns an array of tuples, containing the string `input`, sympy `output` using the provided sympy `func`, and `symbolically` boolean |
| | for calling `compare`. |
| | ''' |
| | return [ |
| | ("1.1", func(1.1), False), |
| | ("6.9", func(6.9), False), |
| | ("3.5", func(3.5), False), |
| | ("8", func(8), False), |
| | ("0", func(0), False), |
| | ("290348E32", func(Rational('290348E32')), False), |
| | ("1237.293894239480234", func(Rational('1237.293894239480234')), False), |
| | ("8623.4592104E-2", func(Rational('8623.4592104E-2')), False), |
| | ("\\pi ", func(pi), False), |
| | ("\\sqrt{100}", func(sqrt(100)), False), |
| | ("12,123.4", func(Rational('12123.4')), False), |
| | ("-9.4", func(-9.4), False), |
| | ("-35.9825", func(-35.9825), False), |
| | ("-\\sqrt{5}", func(-sqrt(5)), False), |
| | ("-324E-3", func(Rational('-324E-3')), False), |
| | ("-0.23", func(-0.23), False), |
| | ("\\frac{1}{2}", func(Rational('1/2')), False), |
| | ("\\frac{6}{2}", func(Rational('6/2')), False), |
| | ("\\frac{9}{5}", func(Rational('9/5')), False), |
| | ("\\frac{-42}{6}", func(Rational('-42/6')), False), |
| | ("-\\frac{325}{3}", func(Rational('-325/3')), False), |
| | ("\\frac{\\pi }{2}", func(pi / 2), False), |
| | ("(1+6)/3", func(Rational(1 + 6, 3)), False), |
| | ("1+6/3", func(1 + Rational('6/3')), False), |
| | ("7*4/5", func(7 * 4 / 5), False), |
| | ("15-2.3", func(15 - Rational('2.3')), False), |
| | ("x", func(x), True), |
| | ("x + y", func(x + y), True), |
| | ("\\frac{9x}{4}", func(9 * x / 4), True), |
| | ("y\\pi", func(y * pi), True), |
| | ("2y-y-y", func(2 * y - y - y), True) |
| | ] |
| |
|
| |
|
| | def compare(actual, expected, symbolically=False): |
| | if symbolically: |
| | assert simplify(actual - expected) == 0 |
| | else: |
| | actual_exp_tree = srepr(actual) |
| | expected_exp_tree = srepr(expected) |
| | try: |
| | assert actual_exp_tree == expected_exp_tree |
| | except Exception: |
| | if isinstance(actual, int) or isinstance(actual, float) or actual.is_number and isinstance(expected, int) or isinstance(expected, float) or expected.is_number: |
| | assert actual == expected or actual - expected == 0 or simplify(actual - expected) == 0 |
| | else: |
| | print('expected_exp_tree = ', expected_exp_tree) |
| | print('actual exp tree = ', actual_exp_tree) |
| | raise |
| |
|
| |
|
| | def assert_equal(latex, expr, variable_values={}, symbolically=False): |
| | parsed = process_sympy(latex, variable_values) |
| | compare(parsed, expr, symbolically) |
| |
|
