def test_set_by_inds_scalar(self):
# Nonempty index array:
pm = GPUPortMapper('/foo[0:3]', np.zeros(3, np.double))
pm.set_by_inds(np.array([0, 1]), 1.0)
assert_array_equal(np.ones(2, np.double), pm.data.get()[0:2])
# Empty index array:
pm = GPUPortMapper('/foo[0:3]', np.zeros(3, np.double))
pm.set_by_inds(np.array([]), 1.0)
assert_array_equal(np.zeros(2, np.double), pm.data.get()[0:2])
def test_set_by_inds_array(self):
# Nonempty index array:
data = np.random.rand(3)
pm = GPUPortMapper('/foo[0:3]', data)
new_data = np.arange(2).astype(np.double)
pm.set_by_inds(np.array([0, 1]), new_data)
assert_array_equal(new_data, pm.data.get()[0:2])
# Empty index array:
data = np.random.rand(3)
pm = GPUPortMapper('/foo[0:3]', data)
new_data = np.arange(2).astype(np.double)
pm.set_by_inds(np.array([], np.int_), new_data)
assert_array_equal(data[0:2], pm.data.get()[0:2])
def test_set_scalar(self):
# Nonempty index array:
data = np.random.rand(3)
pm = GPUPortMapper('/foo[0:3]', data)
new_data = 1.0
ind = np.array([0, 1])
pm.set_by_inds(ind, new_data)
assert_array_equal(np.full(ind.shape, new_data, type(new_data)),
pm.data.get()[0:2])
# Empty index array:
data = np.random.rand(3)
pm = GPUPortMapper('/foo[0:3]', data)
new_data = 1.0
pm.set_by_inds(np.array([], np.int_), new_data)
assert_array_equal(data[0:2], pm.data.get()[0:2])
def test_set_by_inds(self):
data = np.random.rand(3)
pm = GPUPortMapper('/foo[0:3]', data)
new_data = np.arange(2).astype(np.double)
pm.set_by_inds(np.array([0, 1]), new_data)
assert_array_equal(new_data, pm.data.get()[0:2])
