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
|
class AppTestCallMethod:
# The exec hacking is needed to have the code snippets compiled
# by our own compiler, not CPython's
def test_call_method(self):
exec("""if 1:
class C(object):
def m(*args, **kwds):
return args, kwds
sm = staticmethod(m)
cm = classmethod(m)
c = C()
assert c.m() == ((c,), {})
assert c.sm() == ((), {})
assert c.cm() == ((C,), {})
assert c.m(5) == ((c, 5), {})
assert c.sm(6) == ((6,), {})
assert c.cm(7) == ((C, 7), {})
assert c.m(5, x=3) == ((c, 5), {'x': 3})
assert c.m(*range(5)) == ((c, 0, 1, 2, 3, 4), {})
assert c.m(**{'u': 4}) == ((c,), {'u': 4})
""")
def test_call_attribute(self):
exec("""if 1:
class C(object):
def m(*args, **kwds):
return args, kwds
def myfunc(*args):
return args
c = C()
c.m = c.m2 = myfunc
assert c.m() == ()
assert c.m(5, 2) == (5, 2)
assert c.m2(4) == (4,)
assert c.m2("h", "e", "l", "l", "o") == tuple("hello")
""")
def test_call_module(self):
exec("""if 1:
import sys
try:
sys.exit(5)
except SystemExit as e:
assert e.args == (5,)
else:
raise Exception("did not raise?")
""")
def test_custom_getattr(self):
exec("""if 1:
class C(object):
def __getattr__(self, name):
if name == 'bla':
return myfunc
raise AttributeError
def myfunc(*args):
return args
c = C()
assert c.bla(1, 2, 3) == (1, 2, 3)
""", {})
def test_custom_getattribute(self):
exec("""if 1:
class C(object):
def __getattribute__(self, name):
if name == 'bla':
return myfunc
raise AttributeError
def bla(self):
Boom
def myfunc(*args):
return args
c = C()
assert c.bla(1, 2, 3) == (1, 2, 3)
""", {})
def test_builtin(self):
exec("""if 1:
class C(object):
foobar = len
c = C()
assert c.foobar("hello") == 5
""")
def test_attributeerror(self):
exec("""if 1:
assert 5 .__add__(6) == 11
try:
6 .foobar(7)
except AttributeError:
pass
else:
raise Exception("did not raise?")
""")
def test_kwargs(self):
exec("""if 1:
class C(object):
def f(self, a):
return a + 2
assert C().f(a=3) == 5
""")
class TestCallMethod:
def test_space_call_method(self):
space = self.space
w_lst = space.newlist([])
space.call_method(w_lst, 'append', space.w_False)
res = space.int_w(space.call_method(w_lst, '__len__'))
assert res == 1
def test_fallback_case(self):
space = self.space
space.int_w(space.call_method(space.wrap(space.sys),
'getrecursionlimit'))
def test_optimizations_enabled(self):
# check that the callmethod module is really enabled.
from pypy.objspace.std import callmethod
assert (self.space.FrameClass.LOOKUP_METHOD.__func__ ==
callmethod.LOOKUP_METHOD)
assert (self.space.FrameClass.CALL_METHOD.__func__ ==
callmethod.CALL_METHOD)
|
