Copyright update 2022

[modules/shaper.git] / src / ModelAPI / Test / TestSaveOpen2.py

diff --git a/src/ModelAPI/Test/TestSaveOpen2.py b/src/ModelAPI/Test/TestSaveOpen2.py
index 14a97b4fad8db35417daf0d473c3754e7c345a0c..ca8912b300ab0a6df42cf0e71e0b42b4205b7108 100644 (file)
--- a/src/ModelAPI/Test/TestSaveOpen2.py
+++ b/src/ModelAPI/Test/TestSaveOpen2.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2014-2021  CEA/DEN, EDF R&D
+# Copyright (C) 2014-2022  CEA/DEN, EDF R&D
 #
 # This library is free software; you can redistribute it and/or
 # modify it under the terms of the GNU Lesser General Public
@@ -18,6 +18,7 @@
 #
 
 from salome.shaper import model
+from tempfile import TemporaryDirectory
 from ModelAPI import *
 
 model.begin()
@@ -33,36 +34,37 @@ Translation_1 = model.addTranslation(partSet, [model.selection("COMPOUND", "Part
 model.end()
 
 # check save document in a current folder
-aSession = ModelAPI_Session.get()
-aFiles = StringList()
-aSession.save(".", aFiles)
-assert(len(aFiles) == 3)
+with TemporaryDirectory() as tmp_dir:
+  aSession = ModelAPI_Session.get()
+  aFiles = StringList()
+  aSession.save(tmp_dir, aFiles)
+  assert(len(aFiles) == 3)
 
-# check open of the same document
-assert(aSession.load(".") == False) # error because this document is already opened
+  # check open of the same document
+  assert(aSession.load(tmp_dir) == False) # error because this document is already opened
 
-# close the current document
-aSession.closeAll()
+  # close the current document
+  aSession.closeAll()
 
-# open again: it must be correct now
-assert(aSession.load("."))
+  # open again: it must be correct now
+  assert(aSession.load(tmp_dir))
 
-# activate the Part of session
-model.begin()
-partSet = model.moduleDocument()
-assert(partSet.size("Features") == 2)
-aPart = objectToFeature(partSet.object("Features", 0))
-aPartResult = modelAPI_ResultPart(aPart.results()[0])
-aPartResult.activate()
-aPartDoc = aPartResult.partDoc()
-aSession.setActiveDocument(aPartDoc, True)
-model.do()
-# check the field result data
-aFieldFeature = aPartDoc.objectByName("Features", "Field_1")
-aFieldResult = modelAPI_ResultField(objectToFeature(aFieldFeature).results()[0])
-assert(aFieldResult.textLine(0) == "5")
-aSession.setActiveDocument(partSet, True)
-model.do()
-aTranslation = objectToFeature(partSet.objectByName("Features", "Translation_1"))
-partSet.setCurrentFeature(aTranslation, True)
-model.end()
+  # activate the Part of session
+  model.begin()
+  partSet = model.moduleDocument()
+  assert(partSet.size("Features") == 2)
+  aPart = objectToFeature(partSet.object("Features", 0))
+  aPartResult = modelAPI_ResultPart(aPart.results()[0])
+  aPartResult.activate()
+  aPartDoc = aPartResult.partDoc()
+  aSession.setActiveDocument(aPartDoc, True)
+  model.do()
+  # check the field result data
+  aFieldFeature = aPartDoc.objectByName("Features", "Field_1")
+  aFieldResult = modelAPI_ResultField(objectToFeature(aFieldFeature).results()[0])
+  assert(aFieldResult.textLine(0) == "5")
+  aSession.setActiveDocument(partSet, True)
+  model.do()
+  aTranslation = objectToFeature(partSet.objectByName("Features", "Translation_1"))
+  partSet.setCurrentFeature(aTranslation, True)
+  model.end()