fix: class in __init__.py

Tencent · Jul 25, 2023 · 4353c3b · 4353c3b
1 parent 588ac01
commit 4353c3b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/testbase/loader.py b/testbase/loader.py
@@ -143,6 +143,9 @@ def _load(self, testname):
             try:
                 module = __import__(modulename)  # __import__得到的是最外层模块的object
             except Exception as ex:  # pylint: disable=broad-except
+                if i == len(parts) and hasattr(module, parts[-1]):  # 类写在__init__.py的场景下module需返回文件夹
+                    i -= 1
+                    break
                 self._module_errs[modulename] = traceback.format_exc()
                 return
             else: