def test_masked_array(self):
arr = np.ma.array([12, 10, 17], mask=[0, 0, 1], fill_value=42)
dst = np.ma.array([0, 0, 0], mask=False)
req = Request.with_value(arr)
req.writeInto(dst)
assert_array_equal(arr.data, dst.data)
assert_array_equal(arr.mask, dst.mask)
assert arr.get_fill_value() == dst.get_fill_value()
assert_array_equal(np.array([12, 10, 42]), arr.filled())
assert_array_equal(arr.filled(), dst.filled())
def test_notify_finished(self):
cb = mock.Mock()
req = Request.with_value(42)
req.notify_finished(cb)
cb.assert_called_once_with(42)
def test_return_value(self):
req = Request.with_value(None)
assert req.wait() == None
req = Request.with_value(42)
assert req.wait() == 42
