src/FeaturesPlugin/Test/TestCopySubShapes.py

   1 # Copyright (C) 2014-2021  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 # Checks that the Copy feature produces correct results and names if sub-shapes of the same shape are selected
  21
  22 from salome.shaper import model
  23
  24 model.begin()
  25 partSet = model.moduleDocument()
  26 Part_1 = model.addPart(partSet)
  27 Part_1_doc = Part_1.document()
  28 Sketch_1 = model.addSketch(Part_1_doc, model.defaultPlane("XOY"))
  29 SketchLine_1 = Sketch_1.addLine(-25.99630541871921, 19.52832512315271, -16.66748768472907, 29.72783251231528)
  30 SketchLine_2 = Sketch_1.addLine(-16.66748768472907, 29.72783251231528, -12.4384236453202, 17.66256157635469)
  31 SketchConstraintCoincidence_1 = Sketch_1.setCoincident(SketchLine_1.endPoint(), SketchLine_2.startPoint())
  32 SketchLine_3 = Sketch_1.addLine(-12.4384236453202, 17.66256157635469, -25.99630541871921, 19.52832512315271)
  33 SketchConstraintCoincidence_2 = Sketch_1.setCoincident(SketchLine_2.endPoint(), SketchLine_3.startPoint())
  34 SketchConstraintCoincidence_3 = Sketch_1.setCoincident(SketchLine_1.startPoint(), SketchLine_3.endPoint())
  35 model.do()
  36 Extrusion_1 = model.addExtrusion(Part_1_doc, [model.selection("FACE", "Sketch_1/Face-SketchLine_3r-SketchLine_2r-SketchLine_1r")], model.selection(), 10, 0)
  37 Extrusion_1.result().setName("Prism")
  38 Copy_1_objects = [model.selection("FACE", "Prism/Generated_Face&Sketch_1/SketchLine_3"), model.selection("FACE", "Prism/Generated_Face&Sketch_1/SketchLine_2"), model.selection("EDGE", "[Prism/Generated_Face&Sketch_1/SketchLine_1][Prism/To_Face]")]
  39 Copy_1 = model.addCopy(Part_1_doc, Copy_1_objects, 2)
  40 Copy_1.result().setName("Prism_1")
  41 model.end()
  42
  43 assert(Copy_1.feature().results().size() == 6)
  44
  45 for index in range(6):
  46   # name is just incremented
  47   assert(Copy_1.feature().results()[index].data().name() == "Prism_" + str(index + 1))
  48   # type of the shape corresponds to selection: 2 faces then edge
  49   if index%3 == 2:
  50     assert(Copy_1.feature().results()[index].shape().shapeTypeStr() == "EDGE")
  51   else:
  52     assert(Copy_1.feature().results()[index].shape().shapeTypeStr() == "FACE")
  53
  54 model.begin()
  55 model.testHaveNamingSubshapes(Copy_1, model, Part_1_doc)
  56 model.end()
  57
  58 assert(model.checkPythonDump())