def test_series_take(ntake, keep_index):
np.random.seed(0)
nelem = 123
data = np.random.randint(0, 20, nelem)
sr = Series(data)
take_indices = np.random.randint(0, len(sr), ntake)
if keep_index is True:
out = sr.take(take_indices)
np.testing.assert_array_equal(out.to_array(), data[take_indices])
elif keep_index is False:
out = sr.take(take_indices, keep_index=False)
np.testing.assert_array_equal(out.to_array(), data[take_indices])
np.testing.assert_array_equal(out.index.to_array(),
sr.index.to_array())
