class SizedArray_i_test(unittest.TestCase):
def setUp(self):
self.scalar = 23
self.v = np.ones(8, dtype=np.int32)
self.v2 = self.v + 2
self.S = SizedArrayPtr_i(self.v.copy())
self.S2 = SizedArrayPtr_i(self.v2.copy())
def test_deep_copy(self):
v = self.v.copy()
S = self.S.copy(True)
npt.assert_array_equal(v, S.to_numpy())
def test_shallow_copy(self):
v = self.v.copy()
S = self.S.copy(False)
npt.assert_array_equal(v, S.to_numpy())
del self.S
npt.assert_raises(RuntimeError, S.to_numpy)
def test_copy_from(self):
v = self.v.copy()
S = self.S.copy(True)
X = SizedArrayPtr_i().alloc(S.size)
npt.assert_array_equal(v, S.to_numpy())
X.copy_from(S)
npt.assert_array_equal(X.to_numpy(), S.to_numpy())
npt.assert_array_equal(v, S.to_numpy())
def test_pluseq_scalar_from_host(self):
self.S.pluseq_scalar_from_host(self.scalar)
ref = self.v + self.scalar
tst = self.S.to_numpy()
npt.assert_array_equal(ref, tst)
def test_minuseq_scalar_from_host(self):
self.S.minuseq_scalar_from_host(self.scalar)
ref = self.v - self.scalar
tst = self.S.to_numpy()
npt.assert_array_equal(ref, tst)
def test_timeseq_scalar_from_host(self):
self.S.timeseq_scalar_from_host(self.scalar)
ref = self.v * self.scalar
tst = self.S.to_numpy()
npt.assert_array_equal(ref, tst)
def test_shallow_modify_scalar(self):
S = self.S.copy(False)
self.S.minuseq_scalar_from_host(self.scalar)
ref = self.v - self.scalar
tst = S.to_numpy()
npt.assert_array_equal(ref, tst)
def test_pluseq(self):
ref = self.v + self.v2
self.S.pluseq(self.S2)
tst = self.S.to_numpy()
# Preservation
npt.assert_array_equal(self.v2, self.S2.to_numpy())
npt.assert_array_equal(ref, tst)
def test_minuseq(self):
ref = self.v - self.v2
self.S.minuseq(self.S2)
tst = self.S.to_numpy()
# Preservation
npt.assert_array_equal(self.v2, self.S2.to_numpy())
npt.assert_array_equal(ref, tst)
def test_timeseq(self):
ref = self.v * self.v2
self.S.timeseq(self.S2)
tst = self.S.to_numpy()
# Preservation
npt.assert_array_equal(self.v2, self.S2.to_numpy())
npt.assert_array_equal(ref, tst)
def test_roundtrip(self):
s = SizedArrayPtr_i(self.v.copy(), "Round Trip")
ref = self.v
tst = s.to_numpy()
npt.assert_array_equal(ref, tst)
def test_SizedArray2_roundtrip(self):
v2 = np.random.random_integers(0, 24, 50).reshape(
(10, 5)).astype(np.int32)
s = SizedArrayPtr_i(v2, "Round Trip")
ref = v2
tst = s.to_numpy()
npt.assert_array_equal(ref, tst)
class SizedArray_i_test(unittest.TestCase):
def setUp(self):
self.scalar = 23
self.v = np.ones(8, dtype=np.int32)
self.v2 = self.v + 2
self.S = SizedArrayPtr_i(self.v.copy())
self.S2 = SizedArrayPtr_i(self.v2.copy())
def test_deep_copy(self):
v = self.v.copy()
S = self.S.copy(True)
npt.assert_array_equal(v, S.to_numpy())
def test_shallow_copy(self):
v = self.v.copy()
S = self.S.copy(False)
npt.assert_array_equal(v, S.to_numpy())
del self.S
npt.assert_raises(RuntimeError, S.to_numpy)
def test_copy_from(self):
v = self.v.copy()
S = self.S.copy(True)
X = SizedArrayPtr_i().alloc(S.size)
npt.assert_array_equal(v, S.to_numpy())
X.copy_from(S)
npt.assert_array_equal(X.to_numpy(), S.to_numpy())
npt.assert_array_equal(v, S.to_numpy())
def test_pluseq_scalar_from_host(self):
self.S.pluseq_scalar_from_host(self.scalar)
ref = self.v + self.scalar
tst = self.S.to_numpy()
npt.assert_array_equal(ref, tst)
def test_minuseq_scalar_from_host(self):
self.S.minuseq_scalar_from_host(self.scalar)
ref = self.v - self.scalar
tst = self.S.to_numpy()
npt.assert_array_equal(ref, tst)
def test_timeseq_scalar_from_host(self):
self.S.timeseq_scalar_from_host(self.scalar)
ref = self.v * self.scalar
tst = self.S.to_numpy()
npt.assert_array_equal(ref, tst)
def test_shallow_modify_scalar(self):
S = self.S.copy(False)
self.S.minuseq_scalar_from_host(self.scalar)
ref = self.v - self.scalar
tst = S.to_numpy()
npt.assert_array_equal(ref, tst)
def test_pluseq(self):
ref = self.v + self.v2
self.S.pluseq(self.S2)
tst = self.S.to_numpy()
# Preservation
npt.assert_array_equal(self.v2, self.S2.to_numpy())
npt.assert_array_equal(ref, tst)
def test_minuseq(self):
ref = self.v - self.v2
self.S.minuseq(self.S2)
tst = self.S.to_numpy()
# Preservation
npt.assert_array_equal(self.v2, self.S2.to_numpy())
npt.assert_array_equal(ref, tst)
def test_timeseq(self):
ref = self.v * self.v2
self.S.timeseq(self.S2)
tst = self.S.to_numpy()
# Preservation
npt.assert_array_equal(self.v2, self.S2.to_numpy())
npt.assert_array_equal(ref, tst)
def test_roundtrip(self):
s = SizedArrayPtr_i(self.v.copy(), "Round Trip")
ref = self.v
tst = s.to_numpy()
npt.assert_array_equal(ref, tst)
def test_SizedArray2_roundtrip(self):
v2 = np.random.random_integers(0, 24, 50).reshape((10,5)).astype(np.int32)
s = SizedArrayPtr_i(v2, "Round Trip")
ref = v2
tst = s.to_numpy()
npt.assert_array_equal(ref, tst)