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
|
import py, os
from rpython.tool.udir import udir
from rpython.rlib import rvmprof
from rpython.translator.c.test.test_genc import compile
from rpython.rlib.objectmodel import we_are_translated
from rpython.rlib.nonconst import NonConstant
def test_vmprof_execute_code_1():
class MyCode:
pass
try:
rvmprof.register_code_object_class(MyCode, lambda code: 'some code')
except rvmprof.VMProfPlatformUnsupported:
pass
@rvmprof.vmprof_execute_code("xcode1", lambda code, num: code)
def main(code, num):
print num
return 42
def f():
res = main(MyCode(), 5)
assert res == 42
return 0
assert f() == 0
fn = compile(f, [])
assert fn() == 0
def test_vmprof_execute_code_2():
class MyCode:
pass
try:
rvmprof.register_code_object_class(MyCode, lambda code: 'some code')
except rvmprof.VMProfPlatformUnsupported:
pass
class A:
pass
@rvmprof.vmprof_execute_code("xcode2", lambda num, code: code,
result_class=A)
def main(num, code):
print num
return A()
def f():
a = main(7, MyCode())
assert isinstance(a, A)
return 0
assert f() == 0
fn = compile(f, [])
assert fn() == 0
def test_register_code():
class MyCode:
pass
try:
rvmprof.register_code_object_class(MyCode, lambda code: 'some code')
except rvmprof.VMProfPlatformUnsupported as e:
py.test.skip(str(e))
@rvmprof.vmprof_execute_code("xcode1", lambda code, num: code)
def main(code, num):
print num
return 42
def f():
code = MyCode()
rvmprof.register_code(code, lambda code: 'some code')
res = main(code, 5)
assert res == 42
return 0
assert f() == 0
fn = compile(f, [], gcpolicy="minimark")
assert fn() == 0
def test_enable():
class MyCode:
pass
def get_name(code):
return 'py:code:52:x'
try:
rvmprof.register_code_object_class(MyCode, get_name)
except rvmprof.VMProfPlatformUnsupported as e:
py.test.skip(str(e))
@rvmprof.vmprof_execute_code("xcode1", lambda code, num: code)
def main(code, num):
print num
s = 0
for i in range(num):
s += (i << 1)
if s % 2123423423 == 0:
print s
return s
tmpfilename = str(udir.join('test_rvmprof'))
def f():
if NonConstant(False):
# Hack to give os.open() the correct annotation
os.open('foo', 1, 1)
code = MyCode()
rvmprof.register_code(code, get_name)
fd = os.open(tmpfilename, os.O_WRONLY | os.O_CREAT, 0666)
if we_are_translated():
num = 100000000
period = 0.0001
else:
num = 10000
period = 0.9
rvmprof.enable(fd, period)
res = main(code, num)
#assert res == 499999500000
rvmprof.disable()
os.close(fd)
return 0
def check_profile(filename):
from vmprof import read_profile
prof = read_profile(filename)
assert prof.get_tree().name.startswith("py:")
assert prof.get_tree().count
assert f() == 0
assert os.path.exists(tmpfilename)
fn = compile(f, [], gcpolicy="minimark")
assert fn() == 0
try:
import vmprof
except ImportError:
py.test.skip("vmprof unimportable")
else:
check_profile(tmpfilename)
finally:
assert os.path.exists(tmpfilename)
os.unlink(tmpfilename)
|
