test.API/SHAPER/Primitives/TestTorus.py

   1 from salome.shaper import model
   2
   3 model.begin()
   4 partSet = model.moduleDocument()
   5 Part_1 = model.addPart(partSet)
   6 Part_1_doc = Part_1.document()
   7
   8 # Init
   9 Point_1 = model.addPoint(Part_1_doc, 10, 10, 10).result()
  10 Point_2 = model.selection("VERTEX", "pnt")
  11 Axis_1 = model.addAxis(Part_1_doc, 0, 10, 0).result()
  12 Axis_2 = model.selection("EDGE", "edge")
  13
  14 # Parameters
  15 model.addParameter(Part_1_doc, "r", "10")
  16 model.addParameter(Part_1_doc, "ring", "3")
  17
  18 # Tests
  19 Torus_1 = model.addTorus(Part_1_doc, model.selection("VERTEX", "PartSet/Origin"), model.selection("EDGE", "PartSet/OZ"), 15, 3)
  20 Torus_2 = model.addTorus(Part_1_doc, model.selection("VERTEX", "PartSet/Origin"), model.selection("EDGE", "PartSet/OZ"), 5, 10)
  21 Torus_3 = model.addTorus(Part_1_doc, model.selection("VERTEX", "PartSet/Origin"), model.selection("EDGE", "PartSet/OZ"), 0, 3)
  22 Torus_4 = model.addTorus(Part_1_doc, model.selection("VERTEX", "PartSet/Origin"), model.selection("EDGE", "PartSet/OZ"), -10, 3)
  23 Torus_5 = model.addTorus(Part_1_doc, model.selection("VERTEX", "PartSet/Origin"), model.selection("EDGE", "PartSet/OZ"), 15, 0)
  24 Torus_6 = model.addTorus(Part_1_doc, model.selection("VERTEX", "PartSet/Origin"), model.selection("EDGE", "PartSet/OZ"), 15, -3)
  25 Torus_7 = model.addTorus(Part_1_doc, Point_1, model.selection("EDGE", "PartSet/OX"), 12, 7)
  26 Torus_8 = model.addTorus(Part_1_doc, Point_1, Axis_1, 15, 4)
  27 Torus_9 = model.addTorus(Part_1_doc, Point_2, model.selection("EDGE", "PartSet/OZ"), 15, 4)
  28 Torus_10 = model.addTorus(Part_1_doc, model.selection("VERTEX", "PartSet/Origin"), Axis_2, 15, 4)
  29 Torus_11 = model.addTorus(Part_1_doc, model.selection("VERTEX", "PartSet/Origin"), model.selection("EDGE", "PartSet/OZ"), "r", "ring")
  30 Torus_12 = model.addTorus(Part_1_doc, 15, 4)
  31
  32 model.do()
  33 model.end()
  34
  35 # Checks
  36 from GeomAPI import GeomAPI_Shape
  37
  38 model.testNbResults(Torus_1, 1)
  39 model.testNbSubResults(Torus_1, [0])
  40 model.testNbSubShapes(Torus_1, GeomAPI_Shape.SOLID, [1])
  41 model.testNbSubShapes(Torus_1, GeomAPI_Shape.FACE, [1])
  42 model.testHaveNamingFaces(Torus_1, model, Part_1_doc)
  43
  44 model.testNbResults(Torus_7, 1)
  45 model.testNbSubResults(Torus_7, [0])
  46 model.testNbSubShapes(Torus_7, GeomAPI_Shape.SOLID, [1])
  47 model.testNbSubShapes(Torus_7, GeomAPI_Shape.FACE, [1])
  48 model.testHaveNamingFaces(Torus_7, model, Part_1_doc)
  49
  50 model.testNbResults(Torus_8, 1)
  51 model.testNbSubResults(Torus_8, [0])
  52 model.testNbSubShapes(Torus_8, GeomAPI_Shape.SOLID, [1])
  53 model.testNbSubShapes(Torus_8, GeomAPI_Shape.FACE, [1])
  54 model.testHaveNamingFaces(Torus_8, model, Part_1_doc)
  55
  56 model.testNbResults(Torus_11, 1)
  57 model.testNbSubResults(Torus_11, [0])
  58 model.testNbSubShapes(Torus_11, GeomAPI_Shape.SOLID, [1])
  59 model.testNbSubShapes(Torus_11, GeomAPI_Shape.FACE, [1])
  60 model.testHaveNamingFaces(Torus_11, model, Part_1_doc)
  61
  62 model.testNbResults(Torus_12, 1)
  63 model.testNbSubResults(Torus_12, [0])
  64 model.testNbSubShapes(Torus_12, GeomAPI_Shape.SOLID, [1])
  65 model.testNbSubShapes(Torus_12, GeomAPI_Shape.FACE, [1])
  66 model.testHaveNamingFaces(Torus_12, model, Part_1_doc)
  67
  68 model.testNbResults(Torus_2, 0)
  69 assert(Torus_2.feature().error() == "Torus builder :: ring radius is greater than the radius.")
  70
  71 model.testNbResults(Torus_3, 0)
  72 assert(Torus_3.feature().error() == "Torus builder :: radius is negative or null.")
  73
  74 model.testNbResults(Torus_4, 0)
  75 assert(Torus_4.feature().error() == "Torus builder :: radius is negative or null.")
  76
  77 model.testNbResults(Torus_5, 0)
  78 assert(Torus_5.feature().error() == "Torus builder :: ring radius is negative or null.")
  79
  80 model.testNbResults(Torus_6, 0)
  81 assert(Torus_6.feature().error() == "Torus builder :: ring radius is negative or null.")
  82
  83 model.testNbResults(Torus_9, 0)
  84 assert(Torus_9.feature().error() == "Attribute \"base_point\" is not initialized.")
  85
  86 model.testNbResults(Torus_10, 0)
  87 assert(Torus_10.feature().error() == "Attribute \"axis\" is not initialized.")