def test_shuffle_bool_false():
dsi = DatasetIndex(5)
left = dsi._shuffle(shuffle=False)
right = np.arange(5)
assert (left == right).all()
def test_shuffle_bool_callable():
""" Callable 'shuffle' should return order. """
dsi = DatasetIndex(5)
left = dsi._shuffle(shuffle=(lambda _: np.arange(5)))
assert (left == np.arange(5)).all()
def test_shuffle_bool_randomstate():
dsi = DatasetIndex(5)
left = dsi._shuffle(shuffle=np.random.RandomState(13))
right = np.arange(5)
assert (left != right).any()
assert set(left) == set(right)
def test_shuffle_bool_cross():
dsi = DatasetIndex(5)
left = dsi._shuffle(shuffle=np.random.RandomState(13))
right = dsi._shuffle(shuffle=13)
assert (left == right).all()
def test_shuffle_bool_int():
dsi = DatasetIndex(5)
left = dsi._shuffle(shuffle=13)
right = np.arange(5)
assert (left != right).any()
assert set(left) == set(right)
