def test_get_by_name(): """Assert that the Operator class returns operators by name appropriately""" assert Operator.get_by_name("SelectKBest").__class__ == TPOTSelectKBest
def test_operators(): """Assert that the TPOT operators match the output of their sklearn counterparts""" for op in Operator.inheritors(): check_export.description = ("Assert that the TPOT {} operator exports " "as expected".format(op.__name__)) yield check_export, op