X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FModelAPI%2FTest%2FTestSelectionInitialization.py;h=f2ea030f43bec086eb01ecba9abb076473d1765f;hb=77ce6d35ac8d2f0fdaecb4f23e0870bf74e36103;hp=02a40dd405c14b4dd7f71bc6c1702cb4840a7b7a;hpb=d9d68ad55d070600eac1027ac23575616edbe69c;p=modules%2Fshaper.git diff --git a/src/ModelAPI/Test/TestSelectionInitialization.py b/src/ModelAPI/Test/TestSelectionInitialization.py index 02a40dd40..f2ea030f4 100644 --- a/src/ModelAPI/Test/TestSelectionInitialization.py +++ b/src/ModelAPI/Test/TestSelectionInitialization.py @@ -1,4 +1,4 @@ -# Copyright (C) 2014-2019 CEA/DEN, EDF R&D +# Copyright (C) 2014-2024 CEA, EDF # # 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() @@ -29,25 +30,26 @@ Sphere_1 = model.addSphere(Part_1_doc, model.selection("VERTEX", "Point_1"), 10) model.end() # save document in a current folder -aSession = ModelAPI_Session.get() -aFiles = StringList() -aSession.save(".", aFiles) -# close the current document -aSession.closeAll() -# open the saved document -assert(aSession.load(".")) +with TemporaryDirectory() as tmp_dir: + aSession = ModelAPI_Session.get() + aFiles = StringList() + aSession.save(tmp_dir, aFiles) + # close the current document + aSession.closeAll() + # open the saved document + assert(aSession.load(tmp_dir)) -# activate the Part of session -model.begin() -partSet = model.moduleDocument() -assert(partSet.size("Features") == 1) -aPart = objectToFeature(partSet.object("Features", 0)) -aPartResult = modelAPI_ResultPart(aPart.results()[0]) -aPartResult.activate() -aPartDoc = aPartResult.partDoc() -aSession.setActiveDocument(aPartDoc, True) -model.end() + # activate the Part of session + model.begin() + partSet = model.moduleDocument() + assert(partSet.size("Features") == 1) + aPart = objectToFeature(partSet.object("Features", 0)) + aPartResult = modelAPI_ResultPart(aPart.results()[0]) + aPartResult.activate() + aPartDoc = aPartResult.partDoc() + aSession.setActiveDocument(aPartDoc, True) + model.end() -# check the sphere location (it should not be "Origin" - default one) -aSphere = objectToFeature(aPartDoc.objectByName("Features", "Sphere_1")) -assert(aSphere.selection("center_point").namingName() == "Point_1") + # check the sphere location (it should not be "Origin" - default one) + aSphere = objectToFeature(aPartDoc.objectByName("Features", "Sphere_1")) + assert(aSphere.selection("center_point").namingName() == "Point_1")