Skip to content

Commit 6eb9117

Browse files
authored
Added get_pyaedt_app unit test. (#849)
1 parent a1805dd commit 6eb9117

File tree

3 files changed

+12
-3
lines changed

3 files changed

+12
-3
lines changed

_unittest/test_01_Design.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import os
44

55
# Import required modules
6-
from pyaedt import Hfss, Desktop
6+
from pyaedt import Hfss, Desktop, get_pyaedt_app
77
from pyaedt.generic.filesystem import Scratch
88

99
# Setup paths for module imports
@@ -242,3 +242,7 @@ def test_27_odesktop(self):
242242
assert str(type(self.aedtapp.odesktop)) == "<type 'ADesktopWrapper'>"
243243
else:
244244
assert str(type(self.aedtapp.odesktop)) == "<class 'win32com.client.CDispatch'>"
245+
246+
def test_28_get_pyaedt_app(self):
247+
app = get_pyaedt_app(self.aedtapp.project_name, self.aedtapp.design_name)
248+
assert app.design_type == "HFSS"

pyaedt/desktop.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -707,7 +707,7 @@ def design_list(self, project=None):
707707
708708
Returns
709709
-------
710-
str
710+
list
711711
List of the designs.
712712
"""
713713

pyaedt/generic/design_types.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import sys
2+
import re
23

34
try:
45
from pyaedt.hfss3dlayout import Hfss3dLayout
@@ -78,7 +79,11 @@ def get_pyaedt_app(project_name=None, design_name=None):
7879
oProject = main.oDesktop.SetActiveProject(project_name)
7980
if not oProject:
8081
raise AttributeError("No Project Present.")
81-
design_names = [i.GetName() for i in oProject.GetDesigns()]
82+
design_names = []
83+
deslist = list(oProject.GetTopDesignList())
84+
for el in deslist:
85+
m = re.search(r"[^;]+$", el)
86+
design_names.append(m.group(0))
8287
if design_name and design_name not in design_names:
8388
raise AttributeError("Design {} doesn't exists in current Project.".format(design_name))
8489
if not design_name:

0 commit comments

Comments
 (0)