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