Support saving the tnp recalculated file in profile testing

This commit is contained in:
bgbsww
2024-05-09 07:01:31 -04:00
committed by Chris Hennes
parent 4c87e22b7f
commit 5d15df8729
3 changed files with 11 additions and 1 deletions

View File

@@ -15,6 +15,7 @@ SET(Test_SRCS
unittestgui.py
testmakeWireString.py
TestPythonSyntax.py
TestPerf.py
)
SET(TestData_SRCS

View File

@@ -58,6 +58,11 @@ class PerfTestCase(unittest.TestCase):
self.fileList = sys.argv[sys.argv.index("--pass") + 1 :]
else:
raise FileNotFoundError("Must provide filename parameter(s) via --pass")
if "--save" in sys.argv:
self.saveModels = True
self.fileList = sys.argv[sys.argv.index("--save") + 1 :]
else:
self.saveModels = False
if Part.Shape().ElementMapVersion == "":
self.tnp = ""
else:
@@ -91,6 +96,9 @@ class PerfTestCase(unittest.TestCase):
dumpdata = hpy().heap()
dumpdata.stat.dump(self.memfile)
self.memfile.flush()
if self.saveModels:
filenametnp = App.ActiveDocument.Name + self.tnp
App.ActiveDocument.saveAs(filenametnp)
App.closeDocument(doc.Name)
try: