From 6ff179aca21e6c3c5a39c040dd2fd3e224f98a9e Mon Sep 17 00:00:00 2001 From: chenhaoran Date: Tue, 25 Feb 2025 21:23:34 +0800 Subject: [PATCH] refactor: simplify module loading by introducing dynamicLoadModule function --- tests/pytest/test.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/pytest/test.py b/tests/pytest/test.py index 9d0e8651b4..5d1a15961a 100644 --- a/tests/pytest/test.py +++ b/tests/pytest/test.py @@ -149,7 +149,8 @@ if __name__ == "__main__": if is_test_framework: uModule = dynamicLoadModule(fileName) try: - ucase = uModule.TDTestCase() + case_class = getattr(uModule, get_local_classes(uModule)[-1]) + ucase = case_class() tdDnodes.deploy(1,ucase.updatecfgDict) except : tdDnodes.deploy(1,{}) @@ -182,7 +183,8 @@ if __name__ == "__main__": if is_test_framework: uModule = dynamicLoadModule(fileName) try: - ucase = uModule.TDTestCase() + case_class = getattr(uModule, get_local_classes(uModule)[-1]) + ucase = case_class() tdDnodes.deploy(1,ucase.updatecfgDict) except : tdDnodes.deploy(1,{})