def test_operator_type():
"""Assert that TPOT operators return their type, e.g. "Classifier", "Preprocessor" """
assert TPOTSelectKBest().type == "Selector"
def test_operator_export():
"""Assert that a TPOT operator can export properly with a function as a parameter to a classifier"""
export_string = TPOTSelectKBest().export(5)
assert export_string == "SelectKBest(k=5, score_func=f_classif)"