def test_update_instances_on_train():
expect_percentile = [e[:4] for e in expect()]
assert expect() == update_instances(dataset(), {"train_size": 1.0},
"train") # NOQA
assert expect_percentile == update_instances(dataset(),
{"train_size": 0.8},
"train") # NOQA
def test_update_instances_on_test():
assert expect() == update_instances(dataset(), {"test_size": 1.0}, "test")
assert expect() == update_instances(dataset(), {"test_size": 0.8}, "test")
def test_update_instances_on_valid():
assert expect() == update_instances(dataset(), {"valid_size": 1.0},
"valid") # NOQA
assert expect() == update_instances(dataset(), {"valid_size": 0.8},
"valid") # NOQA