src/ModelAPI/Test/TestFolder_Create.py

   1 # Copyright (C) 2014-2020  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 email : webmaster.salome@opencascade.com
  18 #
  19
  20 #=========================================================================
  21 # Initialization of the test
  22 #=========================================================================
  23 from ModelAPI import *
  24 from GeomDataAPI import *
  25
  26 __updated__ = "2017-11-22"
  27
  28 #=========================================================================
  29 # Test 1. Check the Folder is created in PartSet
  30 #=========================================================================
  31 aSession = ModelAPI_Session.get()
  32 aPartSetDoc = aSession.activeDocument()
  33
  34 aSession.startOperation()
  35 aPoint0 = aPartSetDoc.addFeature("Point")
  36 aPoint0Data = aPoint0.data()
  37 assert(aPoint0Data is not None)
  38 geomDataAPI_Point(aPoint0Data.attribute("point3d")).setValue(0., 0., 0.)
  39 aPoint0Data.string("creation_method").setValue("by_xyz")
  40 aSession.finishOperation()
  41
  42 aSession.startOperation()
  43 aPoint1 = aPartSetDoc.addFeature("Point")
  44 aPoint1Data = aPoint1.data()
  45 assert(aPoint1Data is not None)
  46 geomDataAPI_Point(aPoint1Data.attribute("point3d")).setValue(0., 0., 0.)
  47 aPoint1Data.string("creation_method").setValue("by_xyz")
  48 aSession.finishOperation()
  49
  50 assert(aPartSetDoc.size("Features") == 2), "Wrong number of features: {}".format(aPartSetDoc.size("Features"))
  51
  52 # Folder before the feature
  53 aSession.startOperation()
  54 aFolder1 = aPartSetDoc.addFolder(aPoint1)
  55 aSession.finishOperation()
  56
  57 assert(aPartSetDoc.size("Folders") == 1), "Wrong number of folders: {}".format(aPartSetDoc.size("Folders"))
  58 assert(aPartSetDoc.size("Features") == 3), "Wrong number of features: {}".format(aPartSetDoc.size("Features"))
  59 FOLDER_NAME_EXPECTED = "Folder_1"
  60 assert(aFolder1.name() == FOLDER_NAME_EXPECTED), "Actual name '{}', expected '{}'".format(aFolder1.name(), FOLDER_NAME_EXPECTED)
  61
  62 ## Folder at the end of features list
  63 #aSession.startOperation()
  64 #aPartSetDoc.addFolder()
  65 #aSession.finishOperation()
  66 #assert(aPartSetDoc.size("Folders") == 2)
  67
  68 #=========================================================================
  69 # Test 2. Check the Folder is created in a Part
  70 #=========================================================================
  71 aSession.startOperation()
  72 aPart = aSession.moduleDocument().addFeature("Part")
  73 aSession.finishOperation()
  74
  75 # check part is a last feature
  76 aFeaturesList = aSession.moduleDocument().allObjects()
  77 aLast = aFeaturesList[len(aFeaturesList)-1]
  78 assert(aLast.data().isEqual(aPart.data())), "Part is not a last object in the list"
  79
  80 # add point and a folder before it
  81 aPartDoc = aSession.activeDocument()
  82 aSession.startOperation()
  83 aPoint2 = aPartDoc.addFeature("Point")
  84 aPoint2Data = aPoint2.data()
  85 assert(aPoint2Data is not None)
  86 geomDataAPI_Point(aPoint2Data.attribute("point3d")).setValue(0., 0., 0.)
  87 aPoint2Data.string("creation_method").setValue("by_xyz")
  88 aSession.finishOperation()
  89
  90 assert(aPartDoc.size("Features") == 1), "Wrong number of features: {}".format(aPartDoc.size("Features"))
  91
  92 aSession.startOperation()
  93 aFolder2 = aPartDoc.addFolder(aPoint2)
  94 aSession.finishOperation()
  95
  96 assert(aPartDoc.size("Folders") == 1), "Wrong number of folders: {}".format(aPartDoc.size("Folders"))
  97 assert(aPartDoc.size("Features") == 2), "Wrong number of features: {}".format(aPartDoc.size("Features"))
  98 FOLDER_NAME_EXPECTED = "Folder_1"
  99 assert(aFolder2.name() == FOLDER_NAME_EXPECTED), "Actual name '{}', expected '{}'".format(aFolder2.name(), FOLDER_NAME_EXPECTED)