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
|
from __future__ import absolute_import
import sys
import pytest
import os
if os.name != 'posix':
pytest.skip('resource.h only available on unix')
try:
from lib_pypy import resource
except (ImportError, SyntaxError) as e:
pytest.skip(str(e))
def test_getrusage():
x = resource.getrusage(resource.RUSAGE_SELF)
assert len(x) == 16
assert x[0] == x[-16] == x.ru_utime
assert x[1] == x[-15] == x.ru_stime
assert x[2] == x[-14] == x.ru_maxrss
assert x[3] == x[-13] == x.ru_ixrss
assert x[4] == x[-12] == x.ru_idrss
assert x[5] == x[-11] == x.ru_isrss
assert x[6] == x[-10] == x.ru_minflt
assert x[7] == x[-9] == x.ru_majflt
assert x[8] == x[-8] == x.ru_nswap
assert x[9] == x[-7] == x.ru_inblock
assert x[10] == x[-6] == x.ru_oublock
assert x[11] == x[-5] == x.ru_msgsnd
assert x[12] == x[-4] == x.ru_msgrcv
assert x[13] == x[-3] == x.ru_nsignals
assert x[14] == x[-2] == x.ru_nvcsw
assert x[15] == x[-1] == x.ru_nivcsw
for i in range(16):
if i < 2:
expected_type = float
else:
expected_type = int
assert isinstance(x[i], expected_type)
def test_getrlimit():
x = resource.getrlimit(resource.RLIMIT_CPU)
assert isinstance(x, tuple)
assert len(x) == 2
assert isinstance(x[0], int)
assert isinstance(x[1], int)
def test_setrlimit():
# minimal "does not crash" test
x, y = resource.getrlimit(resource.RLIMIT_CPU)
resource.setrlimit(resource.RLIMIT_CPU, (x, y))
# sometimes, x and y are very large (more than 53 bits).
# for these huge values, int(float(x)) > x...
xf = x + (0.2 if x >= 0 else -0.2)
yf = y + (0.3 if y >= 0 else -0.3)
if int(xf) == x and int(yf) == y:
resource.setrlimit(resource.RLIMIT_CPU, (x, y)) # truncated to ints
if sys.platform.startswith("linux") and hasattr(resource, 'prlimit'):
def test_prlimit():
old_limits = resource.getrlimit(resource.RLIMIT_STACK)
assert resource.prlimit(0, resource.RLIMIT_STACK) == old_limits
assert resource.prlimit(os.getpid(), resource.RLIMIT_STACK) == old_limits
# Unchanged
assert resource.getrlimit(resource.RLIMIT_STACK) == old_limits
# Raise the soft limit to match the hard limit
new_limits = (old_limits[1], old_limits[1])
assert resource.prlimit(os.getpid(), resource.RLIMIT_STACK, new_limits) == old_limits
# And make sure it was raised
assert resource.prlimit(0, resource.RLIMIT_STACK, new_limits) == new_limits
assert resource.getrlimit(resource.RLIMIT_STACK) == new_limits
# Change it back
assert resource.prlimit(0, resource.RLIMIT_STACK, old_limits) == new_limits
# And make sure it was changed back
assert resource.prlimit(os.getpid(), resource.RLIMIT_STACK, old_limits) == old_limits
assert resource.getrlimit(resource.RLIMIT_STACK) == old_limits
|
