import unittest
|
from numint.expression import Lark_StandAlone
|
from numint.expression_transformer import ExpressionTransformer
|
|
class MyTestCase(unittest.TestCase):
|
"""
|
tests if the grammer file 'expression.lark' work
|
"""
|
|
def test_prio_product_exponent(self):
|
parser = Lark_StandAlone()
|
tree = parser.parse("x ** -3 * sin(12)")
|
self.assertEqual('''mul
|
exponent
|
var x
|
neg
|
number 3
|
function
|
sin
|
argv
|
number 12
|
''', tree.pretty(indent_str=' '))
|
|
def test_prio_product_exponent_revert(self):
|
parser = Lark_StandAlone()
|
tree = parser.parse("sin(12) * x ** sqrt(4)") # => sin(12) * (x ** sqrt(4))
|
print(f">{tree.pretty()}<")
|
self.assertEqual('''mul
|
function
|
sin
|
argv
|
number 12
|
exponent
|
var x
|
function
|
sqrt
|
argv
|
number 4
|
''', tree.pretty(indent_str=' '))
|
|
def test_use_transform(self):
|
parser = Lark_StandAlone(transformer=ExpressionTransformer())
|
tree = parser.parse("log(-12, 5) * x ** sqrt(4)") # => sin(12) * (x^sqrt(4))
|
self.assertEqual(tree, 'math.log(-12, 5) * x ** math.sqrt(4)')
|
|
|
def test_use_function_in_function(self):
|
parser = Lark_StandAlone(transformer=ExpressionTransformer())
|
tree = parser.parse("tan(radians(x))") # =>
|
self.assertEqual(tree, 'math.tan(math.radians(x))')
|
|
|
|
|
if __name__ == '__main__':
|
unittest.main()
|
