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
|
class AppTestIntOp:
spaceconfig = dict(usemodules=['__pypy__'])
def w_intmask(self, n):
import sys
n &= (sys.maxsize*2+1)
if n > sys.maxsize:
n -= 2*(sys.maxsize+1)
return int(n)
def test_intmask(self):
import sys
assert self.intmask(sys.maxsize) == sys.maxsize
assert self.intmask(sys.maxsize+1) == -sys.maxsize-1
assert self.intmask(-sys.maxsize-2) == sys.maxsize
N = 2 ** 128
assert self.intmask(N+sys.maxsize) == sys.maxsize
assert self.intmask(N+sys.maxsize+1) == -sys.maxsize-1
assert self.intmask(N-sys.maxsize-2) == sys.maxsize
def test_int_add(self):
import sys
from __pypy__ import intop
assert intop.int_add(40, 2) == 42
assert intop.int_add(sys.maxsize, 1) == -sys.maxsize-1
assert intop.int_add(-2, -sys.maxsize) == sys.maxsize
def test_int_sub(self):
import sys
from __pypy__ import intop
assert intop.int_sub(40, -2) == 42
assert intop.int_sub(sys.maxsize, -1) == -sys.maxsize-1
assert intop.int_sub(-2, sys.maxsize) == sys.maxsize
def test_int_mul(self):
import sys
from __pypy__ import intop
assert intop.int_mul(40, -2) == -80
assert intop.int_mul(-sys.maxsize, -sys.maxsize) == (
self.intmask(sys.maxsize ** 2))
def test_int_floordiv(self):
import sys
from __pypy__ import intop
assert intop.int_floordiv(41, 3) == 13
assert intop.int_floordiv(41, -3) == -13
assert intop.int_floordiv(-41, 3) == -13
assert intop.int_floordiv(-41, -3) == 13
assert intop.int_floordiv(-sys.maxsize, -1) == sys.maxsize
assert intop.int_floordiv(sys.maxsize, -1) == -sys.maxsize
def test_int_mod(self):
import sys
from __pypy__ import intop
assert intop.int_mod(41, 3) == 2
assert intop.int_mod(41, -3) == 2
assert intop.int_mod(-41, 3) == -2
assert intop.int_mod(-41, -3) == -2
assert intop.int_mod(-sys.maxsize, -1) == 0
assert intop.int_mod(sys.maxsize, -1) == 0
def test_int_lshift(self):
import sys
from __pypy__ import intop
if sys.maxsize == 2**31-1:
bits = 32
else:
bits = 64
assert intop.int_lshift(42, 3) == 42 << 3
assert intop.int_lshift(0, 3333) == 0
assert intop.int_lshift(1, bits-2) == 1 << (bits-2)
assert intop.int_lshift(1, bits-1) == -sys.maxsize-1 == (-1) << (bits-1)
assert intop.int_lshift(-1, bits-2) == (-1) << (bits-2)
assert intop.int_lshift(-1, bits-1) == -sys.maxsize-1
assert intop.int_lshift(sys.maxsize // 3, 2) == (
self.intmask((sys.maxsize // 3) << 2))
assert intop.int_lshift(-sys.maxsize // 3, 2) == (
self.intmask((-sys.maxsize // 3) << 2))
def test_int_rshift(self):
from __pypy__ import intop
assert intop.int_rshift(42, 3) == 42 >> 3
assert intop.int_rshift(-42, 3) == (-42) >> 3
assert intop.int_rshift(0, 3333) == 0
assert intop.int_rshift(-1, 0) == -1
assert intop.int_rshift(-1, 1) == -1
def test_uint_rshift(self):
import sys
from __pypy__ import intop
if sys.maxsize == 2**31-1:
bits = 32
else:
bits = 64
N = 1 << bits
assert intop.uint_rshift(42, 3) == 42 >> 3
assert intop.uint_rshift(-42, 3) == (N-42) >> 3
assert intop.uint_rshift(0, 3333) == 0
assert intop.uint_rshift(-1, 0) == -1
assert intop.uint_rshift(-1, 1) == sys.maxsize
assert intop.uint_rshift(-1, bits-2) == 3
assert intop.uint_rshift(-1, bits-1) == 1
def test_mulmod(self):
from __pypy__ import intop
assert intop.int_mulmod(9373891, 9832739, 2**31-1) == 1025488209
def test_invalid(self):
from __pypy__ import intop
with raises(ZeroDivisionError):
intop.int_mod(1, 0)
with raises(ZeroDivisionError):
intop.int_floordiv(1, 0)
with raises(ValueError):
intop.int_mulmod(1, 0, 0)
|
