def test_shuffle_breaks_correlations(shuffling_frame):
"""Check shuffler does change covariances"""
while True:
frm = shuffle(shuffling_frame, True)
if (frm != shuffling_frame).values.any():
break
assert (frm.corr() != shuffling_frame.corr()).values.any()
def test_shuffle_retains_averages(shuffling_frame):
"""Check shuffler doesn't change mean values"""
frm = shuffle(shuffling_frame, True)
assert (frm.mean() == shuffling_frame.mean()).values.all()
def test_shuffle_logs_warning(small_shuffling_frame, caplog):
"""Check shuffler raises Warning when shuffle small data set."""
caplog.set_level(logging.INFO)
shuffle(small_shuffling_frame)
assert all(record.levelname == "INFO" for record in caplog.records)
def test_shuffle_returns_nans(small_shuffling_frame):
"""Check shuffler returns nans when shuffling small data sets."""
frm = shuffle(small_shuffling_frame)
assert frm.isna().values.all()
def test_shuffle_logs_message(small_shuffling_frame, caplog):
"""Check shuffler raises message when shuffle small data sets."""
caplog.set_level(logging.INFO)
shuffle(small_shuffling_frame)
assert caplog.records