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
150
151
152
153
154
155
|
from petsc4py import PETSc
import unittest
# --------------------------------------------------------------------
class BaseTestLGMap:
def _mk_idx(self, comm):
comm_size = comm.getSize()
comm_rank = comm.getRank()
lsize = 10
first = lsize * comm_rank
last = first + lsize
if comm_rank > 0:
first -= 1
if comm_rank < (comm_size - 1):
last += 1
return list(range(first, last))
def tearDown(self):
self.lgmap = None
PETSc.garbage_cleanup()
def testGetSize(self):
size = self.lgmap.getSize()
self.assertTrue(size >= 0)
def testGetIndices(self):
size = self.lgmap.getSize()
idx = self.lgmap.getIndices()
self.assertEqual(len(idx), size)
for i, val in enumerate(self.idx):
self.assertEqual(idx[i], val)
def testGetInfo(self):
info = self.lgmap.getInfo()
self.assertEqual(type(info), dict)
if self.lgmap.getComm().getSize() == 1:
self.assertTrue(len(info) == 1)
else:
self.assertTrue(len(info) > 1)
self.assertTrue(len(info) < 4)
def testApply(self):
idxin = list(range(self.lgmap.getSize()))
idxout = self.lgmap.apply(idxin)
self.lgmap.apply(idxin, idxout)
_ = self.lgmap.applyInverse(idxout)
def testApplyIS(self):
is_in = PETSc.IS().createStride(self.lgmap.getSize())
_ = self.lgmap.apply(is_in)
def testProperties(self):
for prop in ('size', 'indices', 'info'):
self.assertTrue(hasattr(self.lgmap, prop))
# --------------------------------------------------------------------
class TestLGMap(BaseTestLGMap, unittest.TestCase):
def setUp(self):
self.idx = self._mk_idx(PETSc.COMM_WORLD)
self.lgmap = PETSc.LGMap().create(self.idx, comm=PETSc.COMM_WORLD)
class TestLGMapIS(BaseTestLGMap, unittest.TestCase):
def setUp(self):
self.idx = self._mk_idx(PETSc.COMM_WORLD)
self.iset = PETSc.IS().createGeneral(self.idx, comm=PETSc.COMM_WORLD)
self.lgmap = PETSc.LGMap().create(self.iset)
def tearDown(self):
self.iset = None
self.lgmap = None
def testSameComm(self):
comm1 = self.lgmap.getComm()
comm2 = self.iset.getComm()
self.assertEqual(comm1, comm2)
# --------------------------------------------------------------------
class TestLGMapBlock(unittest.TestCase):
BS = 3
def setUp(self):
comm = PETSc.COMM_WORLD
comm_size = comm.getSize()
comm_rank = comm.getRank()
lsize = 10
first = lsize * comm_rank
last = first + lsize
if comm_rank > 0:
first -= 1
if comm_rank < (comm_size - 1):
last += 1
self.idx = list(range(first, last))
bs = self.BS
self.lgmap = PETSc.LGMap().create(self.idx, bs, comm=PETSc.COMM_WORLD)
def tearDown(self):
self.lgmap = None
def testGetSize(self):
size = self.lgmap.getSize()
self.assertTrue(size >= 0)
def testGetBlockSize(self):
bs = self.lgmap.getBlockSize()
self.assertEqual(bs, self.BS)
def testGetBlockIndices(self):
size = self.lgmap.getSize()
bs = self.lgmap.getBlockSize()
idx = self.lgmap.getBlockIndices()
self.assertEqual(len(idx), size // bs)
for i, val in enumerate(self.idx):
self.assertEqual(idx[i], val)
def testGetIndices(self):
size = self.lgmap.getSize()
bs = self.lgmap.getBlockSize()
idx = self.lgmap.getIndices()
self.assertEqual(len(idx), size)
for i, val in enumerate(self.idx):
for j in range(bs):
self.assertEqual(idx[i * bs + j], val * bs + j)
def testGetBlockInfo(self):
info = self.lgmap.getBlockInfo()
self.assertEqual(type(info), dict)
if self.lgmap.getComm().getSize() == 1:
self.assertTrue(len(info) == 1)
else:
self.assertTrue(len(info) > 1)
self.assertTrue(len(info) < 4)
def testGetInfo(self):
info = self.lgmap.getInfo()
self.assertEqual(type(info), dict)
if self.lgmap.getComm().getSize() == 1:
self.assertTrue(len(info) == 1)
else:
self.assertTrue(len(info) > 1)
self.assertTrue(len(info) < 4)
# --------------------------------------------------------------------
if __name__ == '__main__':
unittest.main()
|
