def test1d_3(self):
"""
1D, int
"""
dtyp = numpy.int64
# MPI stuff
comm = MPI.COMM_WORLD
rk = comm.Get_rank()
sz = comm.Get_size()
# create the dist array
n = 10
da = distarray.daZeros( (n,), dtyp )
# expose the last element
da.expose( slce=(slice(-1, None, None),), winID='left' )
# set data
da[:] = 100*rk + numpy.array([i for i in range(n)], dtyp)
# access remote data
leftRk = (rk - 1) % sz
print 'proc %d tries to access data from %d' % (rk, leftRk)
leftData = da.get(pe=leftRk, winID='left')
print 'leftData for rank %d = %s' % (rk, str(leftData))
# check
if leftRk < rk:
self.assertEqual(leftData[0], da[-1] - 100)
else:
self.assertEqual(leftData[0], da[-1] + 100*(sz-1))
# free
da.free()
def test0(self):
"""
Test constructors
"""
da = distarray.daZeros( (2,3), numpy.float64 )
da = distarray.daOnes( (2,3), numpy.float64 )
da = distarray.daArray( [1,2,3] )
def test1d_2(self):
"""
1D, float32
"""
dtyp = numpy.float64
# MPI stuff
comm = MPI.COMM_WORLD
rk = comm.Get_rank()
sz = comm.Get_size()
# create the dist array
n = 10
da = distarray.daZeros( (n,), dtyp )
# expose the last element
da.expose( slce=(slice(-1, None, None),), winID='left' )
# set data
da[:] = 100*rk + numpy.array([i for i in range(n)], dtyp)
# access remote data
leftRk = (rk - 1) % sz
print 'proc %d tries to access data from %d' % (rk, leftRk)
leftData = da.get(pe=leftRk, winID='left')
print 'leftData for rank %d = %s' % (rk, str(leftData))
# check
if leftRk < rk:
self.assertEqual(leftData[0], da[-1] - 100)
else:
self.assertEqual(leftData[0], da[-1] + 100*(sz-1))
# free
da.free()
def test2d_1(self):
# create the dist array, the sizes are local to each processor
da = distarray.daZeros( (2,3), numpy.float32 )
# processor rank and number of processes
rk = da.rk
nprocs = da.sz
# expose sub-domains
northSlab = ( slice(-1, None, None), slice(0, None, None) )
da.expose( slce=northSlab, winID='n' )
# set the data
da[:] = rk
# access the remote data, collective operation
northData = da.get( (rk-1) % nprocs, winID='n' )
# check
self.assertEqual(northData.min(), (rk - 1) % nprocs)
self.assertEqual(northData.max(), (rk - 1) % nprocs)
# clean up
da.free()
