src/ModelAPI/Test/TestFolder_Create.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 #=========================================================================
  22 # Initialization of the test
  23 #=========================================================================
  24 from ModelAPI 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 aPoint0Data.real("x").setValue(0.)
  39 aPoint0Data.real("y").setValue(0.)
  40 aPoint0Data.real("z").setValue(0.)
  41 aPoint0Data.string("creation_method").setValue("by_xyz")
  42 aSession.finishOperation()
  43
  44 aSession.startOperation()
  45 aPoint1 = aPartSetDoc.addFeature("Point")
  46 aPoint1Data = aPoint1.data()
  47 assert(aPoint1Data is not None)
  48 aPoint1Data.real("x").setValue(0.)
  49 aPoint1Data.real("y").setValue(0.)
  50 aPoint1Data.real("z").setValue(0.)
  51 aPoint1Data.string("creation_method").setValue("by_xyz")
  52 aSession.finishOperation()
  53
  54 assert(aPartSetDoc.size("Features") == 2), "Wrong number of features: {}".format(aPartSetDoc.size("Features"))
  55
  56 # Folder before the feature
  57 aSession.startOperation()
  58 aFolder1 = aPartSetDoc.addFolder(aPoint1)
  59 aSession.finishOperation()
  60
  61 assert(aPartSetDoc.size("Folders") == 1), "Wrong number of folders: {}".format(aPartSetDoc.size("Folders"))
  62 assert(aPartSetDoc.size("Features") == 3), "Wrong number of features: {}".format(aPartSetDoc.size("Features"))
  63 FOLDER_NAME_EXPECTED = "Folder_1"
  64 assert(aFolder1.name() == FOLDER_NAME_EXPECTED), "Actual name '{}', expected '{}'".format(aFolder1.name(), FOLDER_NAME_EXPECTED)
  65
  66 ## Folder at the end of features list
  67 #aSession.startOperation()
  68 #aPartSetDoc.addFolder()
  69 #aSession.finishOperation()
  70 #assert(aPartSetDoc.size("Folders") == 2)
  71
  72 #=========================================================================
  73 # Test 2. Check the Folder is created in a Part
  74 #=========================================================================
  75 aSession.startOperation()
  76 aPart = aSession.moduleDocument().addFeature("Part")
  77 aSession.finishOperation()
  78
  79 # check part is a last feature
  80 aFeaturesList = aSession.moduleDocument().allObjects()
  81 aLast = aFeaturesList[len(aFeaturesList)-1]
  82 assert(aLast.data().isEqual(aPart.data())), "Part is not a last object in the list"
  83
  84 # add point and a folder before it
  85 aPartDoc = aSession.activeDocument()
  86 aSession.startOperation()
  87 aPoint2 = aPartDoc.addFeature("Point")
  88 aPoint2Data = aPoint2.data()
  89 assert(aPoint2Data is not None)
  90 aPoint2Data.real("x").setValue(0.)
  91 aPoint2Data.real("y").setValue(0.)
  92 aPoint2Data.real("z").setValue(0.)
  93 aPoint2Data.string("creation_method").setValue("by_xyz")
  94 aSession.finishOperation()
  95
  96 assert(aPartDoc.size("Features") == 1), "Wrong number of features: {}".format(aPartDoc.size("Features"))
  97
  98 aSession.startOperation()
  99 aFolder2 = aPartDoc.addFolder(aPoint2)
 100 aSession.finishOperation()
 101
 102 assert(aPartDoc.size("Folders") == 1), "Wrong number of folders: {}".format(aPartDoc.size("Folders"))
 103 assert(aPartDoc.size("Features") == 2), "Wrong number of features: {}".format(aPartDoc.size("Features"))
 104 FOLDER_NAME_EXPECTED = "Folder_1"
 105 assert(aFolder2.name() == FOLDER_NAME_EXPECTED), "Actual name '{}', expected '{}'".format(aFolder2.name(), FOLDER_NAME_EXPECTED)