def test_set_by_inds(self): data = np.random.rand(3) pm = PortMapper('/foo[0:3]', data) new_data = np.arange(2).astype(np.double) pm.set_by_inds([0, 1], new_data) assert_array_equal(new_data, pm.get_by_inds([0, 1]))
def test_get_by_inds(self): data = np.random.rand(3) pm = PortMapper('/foo[0:3]', data) assert_array_equal(data[[0, 1]], pm.get_by_inds([0, 1]))