diff --git a/tests/Observatory/test_Observatory.py b/tests/Observatory/test_Observatory.py index 23261a0d4..4fcd5821d 100644 --- a/tests/Observatory/test_Observatory.py +++ b/tests/Observatory/test_Observatory.py @@ -37,12 +37,17 @@ def setUp(self): self.spec = json.loads(f.read()) modtype = getattr(EXOSIMS.Prototypes.Observatory.Observatory, "_modtype") + whitelist = ["Roman"] pkg = EXOSIMS.Observatory self.allmods = [get_module(modtype)] for loader, module_name, is_pkg in pkgutil.walk_packages( pkg.__path__, pkg.__name__ + "." ): - if not (is_pkg) and ("parallel" not in module_name): + if ( + not (is_pkg) + and ("parallel" not in module_name) + and (module_name.split(".")[-1] not in whitelist) + ): mod = get_module(module_name.split(".")[-1], modtype) self.assertTrue( mod._modtype is modtype, "_modtype mismatch for %s" % mod.__name__