def import_test_class(module_path, class_name):
    for klass in discover(module_path):
        if klass.__name__ == class_name:
            return klass

    raise DiscoveryError(class_name)
dataset/ETHPy150Open Yelp/Testify/testify/test_discovery.py/import_test_class/original