示例#1
0
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")
示例#2
0
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")