def test_kfold_api(): assert_type_error(lambda: kfold(), "Required parameter `nrows` is missing") assert_type_error(lambda: kfold(nrows=5), "Required parameter `nsplits` is missing") assert_type_error(lambda: kfold(nrows=5, nsplits=2, seed=12345), "kfold() got an unexpected keyword argument `seed`") assert_type_error(lambda: kfold(5, 2), "kfold() takes no positional arguments, but 2 were given") assert_type_error(lambda: kfold(nrows=5, nsplits=3.3), "Argument `nsplits` in kfold() should be an integer") assert_type_error(lambda: kfold(nrows=None, nsplits=7), "Argument `nrows` in kfold() should be an integer")
def test_kfold_random_api(): assert_type_error(lambda: kfold_random(), "Required parameter `nrows` is missing") assert_type_error(lambda: kfold_random(nrows=5, seed=12345678), "Required parameter `nsplits` is missing") assert_type_error(lambda: kfold_random(5, 2), "kfold_random() takes no positional arguments, but 2 were given") assert_type_error(lambda: kfold_random(nrows=5, nsplits=3.3), "Argument `nsplits` in kfold_random() should be an integer") assert_type_error(lambda: kfold_random(nrows=None, nsplits=7), "Argument `nrows` in kfold_random() should be an integer") assert_type_error(lambda: kfold_random(nrows=5, nsplits=2, seed="boo"), "Argument `seed` in kfold_random() should be an integer")