demo show image

Hong-Phuc Bui

2024-09-27 8a9933e4decb3b6a57b47b6409474bd476920c24

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
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()