/
test.py
48 lines (41 loc) · 1.26 KB
/
test.py
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
import tempfile
import cmmap
from multiprocessing import Process
def test_file():
fd, filename = tempfile.mkstemp()
f = open(filename, 'w')
data = 'foobarbing'
f.write(data)
f.close()
f = open(filename, 'r+')
m = cmmap.mmap(prot=cmmap.PROT_READ, length=len(data), flags=cmmap.MAP_SHARED, fd=f.fileno())
assert m[:] == data
def test_anonymous_private():
data = 'foobarding'
m = cmmap.mmap(length=10,
prot=cmmap.PROT_READ|cmmap.PROT_WRITE,
flags=cmmap.MAP_ANONYMOUS|cmmap.MAP_PRIVATE)
m[:] = data
assert m[:] == data
def test_anonymous_shared():
data = 'foobarding'
m = cmmap.mmap(length=10,
prot=cmmap.PROT_READ|cmmap.PROT_WRITE,
flags=cmmap.MAP_ANONYMOUS|cmmap.MAP_SHARED)
m[:] = data
assert m[:] == data
def f():
assert m[:] == data
m[0] = 'd'
proc = Process(target=f)
proc.start()
proc.join()
assert m[:] == 'doobarding'
def test_anonymous_private():
data = 'foobarding'
m = cmmap.mmap(length=10,
prot=cmmap.PROT_READ|cmmap.PROT_WRITE,
flags=cmmap.MAP_ANONYMOUS|cmmap.MAP_PRIVATE,
buffer=False)
m[:] = data
assert m[:] == data