MCPcopy Create free account
hub / github.com/mpi4py/mpi4py / check3

Method check3

test/test_msgspec.py:1195–1227  ·  view source on GitHub ↗
(self, z, s, r, typecode)

Source from the content-addressed store, hash-verified

1193 self.assertEqual(a, b)
1194
1195 def check3(self, z, s, r, typecode):
1196 del typecode
1197 size = len(r)
1198 for count in range(size):
1199 r[:] = z
1200 Alltoallv([s, count], [r, count])
1201 for i in range(count):
1202 self.assertEqual(r[i], s[i])
1203 for i in range(count, size):
1204 self.assertEqual(r[i], z[0])
1205 for count in range(size):
1206 r[:] = z
1207 Alltoallv(
1208 [s, (count, None)],
1209 [r, (count, None)],
1210 )
1211 for i in range(count):
1212 self.assertEqual(r[i], s[i])
1213 for i in range(count, size):
1214 self.assertEqual(r[i], z[0])
1215 for disp in range(size):
1216 for count in range(size - disp):
1217 r[:] = z
1218 Alltoallv(
1219 [s, ([count], [disp])],
1220 [r, ([count], [disp])],
1221 )
1222 for i in range(disp):
1223 self.assertEqual(r[i], z[0])
1224 for i in range(disp, disp + count):
1225 self.assertEqual(r[i], s[i])
1226 for i in range(disp + count, size):
1227 self.assertEqual(r[i], z[0])
1228
1229 def check4(self, z, s, r, typecode):
1230 datatype = typemap(typecode)

Callers

nothing calls this directly

Calls 1

AlltoallvFunction · 0.85

Tested by

no test coverage detected