Python maybe_fail_import Examples

Example #1

File: test_magics.py Project: shollyman/python-bigquery

def missing_grpcio_lib():
    """Provide a patcher that can make the gapic library import to fail."""

    def fail_if(name, globals, locals, fromlist, level):
        # NOTE: *very* simplified, assuming a straightforward absolute import
        return "gapic_v1" in name or (fromlist is not None and "gapic_v1" in fromlist)

    return maybe_fail_import(predicate=fail_if)

Example #2

File: test_magics.py Project: beittatt/cloud-python

def missing_bq_storage():
    """Provide a patcher that can make the bigquery storage import to fail."""
    def fail_if(name, globals, locals, fromlist, level):
        # NOTE: *very* simplified, assuming a straightforward absolute import
        return "bigquery_storage_v1beta1" in name or (
            fromlist is not None and "bigquery_storage_v1beta1" in fromlist)

    return maybe_fail_import(predicate=fail_if)