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
|
from pymol import testing
from pymol import constants
class TestConstants(testing.PyMOLTestCase):
def test_safe_alpha_list_eval(self):
f = constants.safe_alpha_list_eval
# expected use case
self.assertEqual(f('[blue, red, green]'), ["blue", "red", "green"])
# single item
self.assertEqual(f('[AB]'), ["AB"])
# trailing comma
self.assertEqual(f('[A,]'), ["A"])
# tuple
self.assertEqual(f("('A',)"), ('A',))
# single value
self.assertEqual(f("('A')"), 'A')
self.assertEqual(f("'A'"), 'A')
# with quotes
self.assertEqual(f('["AB", \'cd\']'), ["AB", "cd"])
# str, int, non-alnum characters
self.assertEqual(f('[A, 1, 2+3, B:C, D E, "F G"]'),
["A", 1, 23, "BC", "DE", "FG"])
@testing.requires_version('1.6')
def test_safe_eval(self):
f = constants.safe_eval
self.assertEqual(f('[AB]'), ["AB"])
self.assertEqual(f('[A,]'), ["A"])
self.assertEqual(f("('A',)"), ('A',))
self.assertEqual(f("('A')"), 'A')
self.assertEqual(f("'A'"), 'A')
self.assertEqual(f('["AB", \'cd\']'), ["AB", "cd"])
self.assertEqual(f('[A, 1, 2+3, "F G"]'), ["A", 1, 5, "F G"])
self.assertEqual(f('()'), ())
self.assertEqual(f('[]'), [])
self.assertEqual(f('{}'), {})
self.assertEqual(f('{foo: bar}'), {"foo": "bar"})
self.assertEqual(f('1 + 2'), 3)
with self.assertRaises(TypeError):
f('__import__("os").times()')
|
