From 2edba36de7eb2987549943e37274cc7616fdde6b Mon Sep 17 00:00:00 2001 From: Aviv Palivoda Date: Thu, 4 Jul 2019 23:58:01 +0300 Subject: [PATCH] Allow skipping of unknown modules --- tests/not_impl_gen.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/not_impl_gen.py b/tests/not_impl_gen.py index 4689eca867..7e0c029e9f 100644 --- a/tests/not_impl_gen.py +++ b/tests/not_impl_gen.py @@ -60,6 +60,12 @@ def gen_methods(header, footer, output): output.write("}\n\n") output.write(footer.read()) +def get_module_methods(name): + try: + return set(dir(__import__(name))) if name not in ("this", "antigravity") else None + except ModuleNotFoundError: + return None + def gen_modules(header, footer, output): output.write(header.read()) @@ -70,9 +76,7 @@ def gen_modules(header, footer, output): mod.name, # check name b/c modules listed have side effects on import, # e.g. printing something or opening a webpage - set(dir(__import__(mod.name))) - if mod.name not in ("this", "antigravity") - else None, + get_module_methods(mod.name) ), pkgutil.iter_modules(), )