def check_op(self, op):
"""Check unary operation for success.
Check the one- and two-arg ufunc versions as well as the method
version attached to a LocalArray.
"""
x = da.ones((16, 16), dist=("b", "n"), comm=self.comm)
y = da.ones((16, 16), dist=("b", "n"), comm=self.comm)
with warnings.catch_warnings():
warnings.simplefilter("ignore", category=RuntimeWarning)
result0 = op(x, casting="unsafe") # standard form
op(x, y=y, casting="unsafe") # two-arg form
assert_array_equal(result0.local_array, y.local_array)
def test_ones(self):
size = self.get_comm_size()
nrows = size * 3
a = dla.ones((nrows, 20), comm=self.comm)
expected = np.ones((nrows / size, 20))
assert_array_equal(a.local_array, expected)
