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 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209
|
"""Test the exec statement functionality.
New for PyPy - Could be incorporated into CPython regression tests.
"""
from rpython.tool.udir import udir
class AppTestExecStmt:
def test_string(self):
g = {}
l = {}
exec("a = 3", g, l)
assert l['a'] == 3
def test_localfill(self):
g = {}
exec("a = 3", g)
assert g['a'] == 3
def test_builtinsupply(self):
g = {}
exec("pass", g)
assert '__builtins__' in g
def test_invalidglobal(self):
def f():
exec('pass', 1)
raises(TypeError, f)
def test_invalidlocal(self):
def f():
exec('pass', {}, 2)
raises(TypeError, f)
def test_codeobject(self):
co = compile("a = 3", '<string>', 'exec')
g = {}
l = {}
exec(co, g, l)
assert l['a'] == 3
def test_implicit(self):
a = 4
exec("a = 3")
assert a == 3
def test_tuplelocals(self):
g = {}
l = {}
exec("a = 3", g, l)
assert l['a'] == 3
def test_tupleglobals(self):
g = {}
exec("a = 3", g)
assert g['a'] == 3
def test_exceptionfallthrough(self):
def f():
exec('raise TypeError', {})
raises(TypeError, f)
def test_global_stmt(self):
g = {}
l = {}
co = compile("global a; a=5", '', 'exec')
#import dis
#dis.dis(co)
exec(co, g, l)
assert l == {}
assert g['a'] == 5
def test_specialcase_free_load(self):
exec("""if 1:
def f():
exec('a=3')
return a
raises(NameError, f)\n""")
def test_specialcase_free_load2(self):
exec("""if 1:
def f(a):
exec('a=3')
return a
x = f(4)\n""")
assert eval("x") == 3
def test_nested_names_are_not_confused(self):
def get_nested_class():
method_and_var = "var"
class Test(object):
def method_and_var(self):
return "method"
def test(self):
return method_and_var
def actual_global(self):
return str("global")
def str(self):
return str(self)
return Test()
t = get_nested_class()
assert t.actual_global() == "global"
assert t.test() == 'var'
assert t.method_and_var() == 'method'
def test_exec_load_name(self):
d = {'x': 2}
exec("""if 1:
def f():
save = x
exec("x=3")
return x,save
\n""", d)
res = d['f']()
assert res == (2, 2)
def test_space_bug(self):
d = {}
exec("x=5 ", d)
assert d['x'] == 5
def test_synerr(self):
def x():
exec("1 2")
raises(SyntaxError, x)
def test_mapping_as_locals(self):
class M(object):
def __getitem__(self, key):
return key
def __setitem__(self, key, value):
self.result[key] = value
def setdefault(self, key, value):
assert key == '__builtins__'
m = M()
m.result = {}
exec("x=m", {}, m)
assert m.result == {'x': 'm'}
try:
exec("y=n", m)
except TypeError:
pass
else:
assert False, 'Expected TypeError'
raises(TypeError, eval, "m", m)
def test_filename(self):
try:
exec("'unmatched_quote")
except SyntaxError as msg:
assert msg.filename == '<string>'
try:
eval("'unmatched_quote")
except SyntaxError as msg:
assert msg.filename == '<string>'
def test_exec_and_name_lookups(self):
ns = {}
exec("""def f():
exec('x=1', globals())
return x\n""", ns)
f = ns['f']
try:
res = f()
except NameError as e: # keep py.test from exploding confused
raise e
assert res == 1
def test_exec_unicode(self):
# 's' is a bytes string
s = b"x = '\xd0\xb9\xd1\x86\xd1\x83\xd0\xba\xd0\xb5\xd0\xbd'"
# 'u' is a unicode
u = s.decode('utf-8')
ns = {}
exec(u, ns)
x = ns['x']
assert len(x) == 6
assert ord(x[0]) == 0x0439
assert ord(x[1]) == 0x0446
assert ord(x[2]) == 0x0443
assert ord(x[3]) == 0x043a
assert ord(x[4]) == 0x0435
assert ord(x[5]) == 0x043d
def test_eval_unicode(self):
u = "'%s'" % chr(0x1234)
v = eval(u)
assert v == chr(0x1234)
def test_compile_bytes(self):
s = b"x = '\xd0\xb9\xd1\x86\xd1\x83\xd0\xba\xd0\xb5\xd0\xbd'"
c = compile(s, '<input>', 'exec')
ns = {}
exec(c, ns)
x = ns['x']
assert len(x) == 6
assert ord(x[0]) == 0x0439
def test_issue3297(self):
c = compile("a, b = '\U0001010F', '\\U0001010F'", "dummy", "exec")
d = {}
exec(c, d)
assert d['a'] == d['b']
assert len(d['a']) == len(d['b'])
assert ascii(d['a']) == ascii(d['b'])
|