Exemplos de define_holdout_set em Python

Exemplo n.º 1

Arquivo: test_environment.py Projeto: databill86/hyperparameter_hunter

def test_define_holdout_set_str(monkeypatch):
    dummy_df = pd.DataFrame(dict(a=[0, 1], b=[2, 3]))

    # noinspection PyUnusedLocal
    def mock_pandas_read_csv(*args, **kwargs):
        return dummy_df

    monkeypatch.setattr(pd, "read_csv", mock_pandas_read_csv)
    assert define_holdout_set(pd.DataFrame(dict(a=[4], b=[5])), "foo", "x")[1].equals(dummy_df)

Exemplo n.º 2

Arquivo: test_environment.py Projeto: yyfyyfgong/find_best_hyperparameters

def test_define_holdout_set_value_error(holdout_set):
    with pytest.raises(ValueError, match="Mismatched columns.*"):
        define_holdout_set(pd.DataFrame(dict(a=[0, 1], b=[2, 3])),
                           pd.DataFrame(holdout_set), "x")

Exemplo n.º 3

Arquivo: test_environment.py Projeto: yyfyyfgong/find_best_hyperparameters

def test_define_holdout_set_file_not_found_error():
    with pytest.raises(FileNotFoundError):
        define_holdout_set(pd.DataFrame(), "foo", "target")

Exemplo n.º 4

Arquivo: test_environment.py Projeto: yyfyyfgong/find_best_hyperparameters

def test_define_holdout_set_type_error(holdout_set):
    with pytest.raises(
            TypeError,
            match="holdout_set must be None, DataFrame, callable, or str,.*"):
        define_holdout_set(pd.DataFrame(), holdout_set, "target")