def test_values(self):
if self.comm.Get_rank() == 0:
assert_array_equal(np.arange(20), self.larr.ndarray)
elif self.comm.Get_rank() == 1:
assert_array_equal(np.arange(30), self.larr.ndarray)
larr = LocalArray.from_distarray(comm=self.comm, obj=self.larr)
if self.comm.Get_rank() == 0:
assert_array_equal(np.arange(20), larr.ndarray)
elif self.comm.Get_rank() == 1:
assert_array_equal(np.arange(30), larr.ndarray)
def test_values(self):
if self.comm.Get_rank() == 0:
assert_array_equal(np.arange(20), self.larr.ndarray)
elif self.comm.Get_rank() == 1:
assert_array_equal(np.arange(30), self.larr.ndarray)
larr = LocalArray.from_distarray(comm=self.comm, obj=self.larr)
if self.comm.Get_rank() == 0:
assert_array_equal(np.arange(20), larr.ndarray)
elif self.comm.Get_rank() == 1:
assert_array_equal(np.arange(30), larr.ndarray)
def test_round_trip_elements(self):
larr = LocalArray.from_distarray(comm=self.comm, obj=self.larr)
if self.comm.Get_rank() == 0:
idx = (0,) * larr.ndarray.ndim
larr.ndarray[idx] = 99
assert_array_equal(larr.ndarray, self.larr.ndarray)
def test_round_trip_equality_from_dict(self):
larr = LocalArray.from_distarray(comm=self.comm, obj=self.larr.__distarray__())
self.assert_round_trip_equality(larr)
def test_round_trip_elements(self):
larr = LocalArray.from_distarray(comm=self.comm, obj=self.larr)
if self.comm.Get_rank() == 0:
idx = (0, ) * larr.ndarray.ndim
larr.ndarray[idx] = 99
assert_array_equal(larr.ndarray, self.larr.ndarray)
def test_round_trip_equality_from_dict(self):
larr = LocalArray.from_distarray(comm=self.comm,
obj=self.larr.__distarray__())
self.assert_round_trip_equality(larr)
