src/ModelAPI/Test/TestSelectionInitialization.py

   1 ## Copyright (C) 2014-2017  CEA/DEN, EDF R&D
   2 ##
   3 ## This library is free software; you can redistribute it and/or
   4 ## modify it under the terms of the GNU Lesser General Public
   5 ## License as published by the Free Software Foundation; either
   6 ## version 2.1 of the License, or (at your option) any later version.
   7 ##
   8 ## This library is distributed in the hope that it will be useful,
   9 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 ## Lesser General Public License for more details.
  12 ##
  13 ## You should have received a copy of the GNU Lesser General Public
  14 ## License along with this library; if not, write to the Free Software
  15 ## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  16 ##
  17 ## See http:##www.salome-platform.org/ or
  18 ## email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
  19 ##
  20
  21 # -*- coding: utf-8 -*-
  22
  23 from salome.shaper import model
  24 from ModelAPI import *
  25
  26 model.begin()
  27 partSet = model.moduleDocument()
  28 Part_1 = model.addPart(partSet)
  29 Part_1_doc = Part_1.document()
  30 Point_2 = model.addPoint(Part_1_doc, 1, 2, 3)
  31 Sphere_1 = model.addSphere(Part_1_doc, model.selection("VERTEX", "Point_1"), 10)
  32 model.end()
  33
  34 # save document in a current folder
  35 aSession = ModelAPI_Session.get()
  36 aFiles = StringList()
  37 aSession.save(".", aFiles)
  38 # close the current document
  39 aSession.closeAll()
  40 # open the saved document
  41 assert(aSession.load("."))
  42
  43 # activate the Part of session
  44 model.begin()
  45 partSet = model.moduleDocument()
  46 assert(partSet.size("Features") == 1)
  47 aPart = objectToFeature(partSet.object("Features", 0))
  48 aPartResult = modelAPI_ResultPart(aPart.results()[0])
  49 aPartResult.activate()
  50 aPartDoc = aPartResult.partDoc()
  51 aSession.setActiveDocument(aPartDoc, True)
  52 model.end()
  53
  54 # check the sphere location (it should not be "Origin" - default one)
  55 aSphere = objectToFeature(aPartDoc.objectByName("Features", "Sphere_1"))
  56 assert(aSphere.selection("center_point").namingName() == "Point_1")