

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