/
test_util.py
44 lines (37 loc) · 1.46 KB
/
test_util.py
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
import unittest2 as unittest
from pyrap.util import substitute
def f1(arg):
a = 3
s = substitute('subs as $a $arg', locals=locals())
print(a, arg, s)
class TestUtil(unittest.TestCase):
def test_util(self):
a = 1
b = 2
p = "$((a+b)*(a+b))"
s = substitute(p, locals=locals())
print("a=%d, b=%d, %s => %s" % (a, b, p, s))
f1(23)
f1('xyz')
def test_substitute(self):
a = 2
b = 3
c = "xyz"
d1 = True
s1 = (1, 2, 3)
s2 = ['ab', 'cde', 'f', 'ghij']
self.assertTrue(substitute('$a $b $c $d1') == '2 3 "xyz" T')
self.assertTrue(substitute('$(a) $(b) $(c) $(d1)') == '2 3 "xyz" T')
self.assertTrue(substitute('$b $0 $a "$a" $b') == '3 $0 2 "$a" 3')
self.assertTrue(substitute('$(a+b)') == '5')
self.assertTrue(substitute('$((a+b)*(a+b))') == '25')
self.assertTrue(substitute('$((a+b)*(a+c))') == '$((a+b)*(a+c))')
self.assertTrue(substitute('"$(a+b)"') == '"$(a+b)"')
self.assertTrue(substitute('\\$(a+b) \\\\$a \\$a') ==
'\\$(a+b) \\\\2 \\$a')
self.assertTrue(substitute('$(a+b)+$a') == '5+2')
self.assertTrue(substitute('$((a+b)+a)') == '7')
self.assertTrue(substitute('$((a+b)*(a+b))') == '25')
self.assertTrue(substitute('$(len("ab cd( de"))') == '9')
self.assertTrue(substitute(
' $s1 $s2 ') == ' [1,2,3] ["ab","cde","f","ghij"] ')