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
|
import sys
from mpi4py import MPI
import mpiunittest as unittest
try:
from sys import getrefcount
except ImportError:
class getrefcount(object):
def __init__(self, arg):
pass
def __eq__(self, other):
return True
def __add__(self, other):
return self
def __sub__(self, other):
return self
def memzero(m):
n = len(m)
if n == 0: return
try:
zero = '\0'.encode('ascii')
m[0] = zero
except TypeError:
zero = 0
m[0] = zero
for i in range(n):
m[i] = zero
class BaseTestWin(object):
COMM = MPI.COMM_NULL
INFO = MPI.INFO_NULL
def setUp(self):
try:
self.mpi_memory = MPI.Alloc_mem(10)
self.memory = self.mpi_memory
memzero(self.memory)
except MPI.Exception:
from array import array
self.mpi_memory = None
self.memory = array('B',[0]*10)
refcnt = getrefcount(self.memory)
self.WIN = MPI.Win.Create(self.memory, 1, self.INFO, self.COMM)
if type(self.memory).__name__ == 'buffer':
self.assertEqual(getrefcount(self.memory), refcnt+1)
else:
if sys.version_info[:3] < (3, 3):
self.assertEqual(getrefcount(self.memory), refcnt)
else:
self.assertEqual(getrefcount(self.memory), refcnt+1)
def tearDown(self):
refcnt = getrefcount(self.memory)
self.WIN.Free()
if type(self.memory).__name__ == 'buffer':
self.assertEqual(getrefcount(self.memory), refcnt-1)
else:
if sys.version_info[:3] < (3, 3):
self.assertEqual(getrefcount(self.memory), refcnt)
else:
self.assertEqual(getrefcount(self.memory), refcnt-1)
if self.mpi_memory:
MPI.Free_mem(self.mpi_memory)
def testGetMemory(self):
memory = self.WIN.memory
pointer = MPI.Get_address(memory)
length = len(memory)
base, size, dunit = self.WIN.attrs
self.assertEqual(size, length)
self.assertEqual(dunit, 1)
self.assertEqual(base, pointer)
def testAttributes(self):
cgroup = self.COMM.Get_group()
wgroup = self.WIN.Get_group()
grpcmp = MPI.Group.Compare(cgroup, wgroup)
cgroup.Free()
wgroup.Free()
self.assertEqual(grpcmp, MPI.IDENT)
base, size, unit = self.WIN.attrs
self.assertEqual(size, len(self.memory))
self.assertEqual(unit, 1)
self.assertEqual(base, MPI.Get_address(self.memory))
def testGetAttr(self):
base = MPI.Get_address(self.memory)
size = len(self.memory)
unit = 1
self.assertEqual(size, self.WIN.Get_attr(MPI.WIN_SIZE))
self.assertEqual(unit, self.WIN.Get_attr(MPI.WIN_DISP_UNIT))
self.assertEqual(base, self.WIN.Get_attr(MPI.WIN_BASE))
def testGetSetErrhandler(self):
for ERRHANDLER in [MPI.ERRORS_ARE_FATAL, MPI.ERRORS_RETURN,
MPI.ERRORS_ARE_FATAL, MPI.ERRORS_RETURN,]:
errhdl_1 = self.WIN.Get_errhandler()
self.assertNotEqual(errhdl_1, MPI.ERRHANDLER_NULL)
self.WIN.Set_errhandler(ERRHANDLER)
errhdl_2 = self.WIN.Get_errhandler()
self.assertEqual(errhdl_2, ERRHANDLER)
errhdl_2.Free()
self.assertEqual(errhdl_2, MPI.ERRHANDLER_NULL)
self.WIN.Set_errhandler(errhdl_1)
errhdl_1.Free()
self.assertEqual(errhdl_1, MPI.ERRHANDLER_NULL)
def testGetSetName(self):
try:
name = self.WIN.Get_name()
self.WIN.Set_name('mywin')
self.assertEqual(self.WIN.Get_name(), 'mywin')
self.WIN.Set_name(name)
self.assertEqual(self.WIN.Get_name(), name)
except NotImplementedError:
pass
class TestWinSelf(BaseTestWin, unittest.TestCase):
COMM = MPI.COMM_SELF
class TestWinWorld(BaseTestWin, unittest.TestCase):
COMM = MPI.COMM_WORLD
try:
w = MPI.Win.Create(MPI.BOTTOM, 1, MPI.INFO_NULL, MPI.COMM_SELF).Free()
except NotImplementedError:
del BaseTestWin, TestWinSelf, TestWinWorld
_name, _version = MPI.get_vendor()
if _name == 'Open MPI':
if _version < (1, 4, 0):
if MPI.Query_thread() > MPI.THREAD_SINGLE:
del TestWinWorld
elif _name == 'MPICH2':
if 'win' in sys.platform:
del BaseTestWin.testAttributes
if __name__ == '__main__':
unittest.main()
|
