예제 #1
0
def test_columnar_convert_disallow_features():
    converter = ColumnarConverter("some_name", "foo", None, {}, {}, False, {})
    df = _EMPTY_DF.assign(a=1)
    with pytest.raises(ValueError,
                       match="expected zero feature columns, found 'a'"):
        shared, type_starts, features = converter.convert(df)
예제 #2
0
def test_columnar_convert_type_default():
    converter = ColumnarConverter("some_name", "foo", {}, {}, False)
    shared, features = converter.convert(_EMPTY_DF)
    assert "foo" in shared
    assert "foo" in features