diff --git a/_unittest/test_01_Design.py b/_unittest/test_01_Design.py index f63b84b323e..0799560cb7b 100644 --- a/_unittest/test_01_Design.py +++ b/_unittest/test_01_Design.py @@ -3,7 +3,7 @@ import os # Import required modules -from pyaedt import Hfss, Desktop +from pyaedt import Hfss, Desktop, get_pyaedt_app from pyaedt.generic.filesystem import Scratch # Setup paths for module imports @@ -242,3 +242,7 @@ def test_27_odesktop(self): assert str(type(self.aedtapp.odesktop)) == "" else: assert str(type(self.aedtapp.odesktop)) == "" + + def test_28_get_pyaedt_app(self): + app = get_pyaedt_app(self.aedtapp.project_name, self.aedtapp.design_name) + assert app.design_type == "HFSS" diff --git a/pyaedt/desktop.py b/pyaedt/desktop.py index 1552b46838d..3583a59cb4e 100644 --- a/pyaedt/desktop.py +++ b/pyaedt/desktop.py @@ -707,7 +707,7 @@ def design_list(self, project=None): Returns ------- - str + list List of the designs. """ diff --git a/pyaedt/generic/design_types.py b/pyaedt/generic/design_types.py index a8c51c6f418..0b9d738dbb0 100644 --- a/pyaedt/generic/design_types.py +++ b/pyaedt/generic/design_types.py @@ -1,4 +1,5 @@ import sys +import re try: from pyaedt.hfss3dlayout import Hfss3dLayout @@ -78,7 +79,11 @@ def get_pyaedt_app(project_name=None, design_name=None): oProject = main.oDesktop.SetActiveProject(project_name) if not oProject: raise AttributeError("No Project Present.") - design_names = [i.GetName() for i in oProject.GetDesigns()] + design_names = [] + deslist = list(oProject.GetTopDesignList()) + for el in deslist: + m = re.search(r"[^;]+$", el) + design_names.append(m.group(0)) if design_name and design_name not in design_names: raise AttributeError("Design {} doesn't exists in current Project.".format(design_name)) if not design_name: