Salome HOME
Merge remote-tracking branch 'origin/cbr/export_to_geom_via_xao'
[modules/shaper.git] / 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
31 model.do()
32 model.end()
33
34 # Checks
35 from GeomAPI import GeomAPI_Shape
36
37 model.testNbResults(Torus_1, 1)
38 model.testNbSubResults(Torus_1, [0])
39 model.testNbSubShapes(Torus_1, GeomAPI_Shape.SOLID, [1])
40 model.testNbSubShapes(Torus_1, GeomAPI_Shape.FACE, [1])
41 model.testHaveNamingFaces(Torus_1, model, Part_1_doc)
42
43 model.testNbResults(Torus_7, 1)
44 model.testNbSubResults(Torus_7, [0])
45 model.testNbSubShapes(Torus_7, GeomAPI_Shape.SOLID, [1])
46 model.testNbSubShapes(Torus_7, GeomAPI_Shape.FACE, [1])
47 model.testHaveNamingFaces(Torus_7, model, Part_1_doc)
48
49 model.testNbResults(Torus_8, 1)
50 model.testNbSubResults(Torus_8, [0])
51 model.testNbSubShapes(Torus_8, GeomAPI_Shape.SOLID, [1])
52 model.testNbSubShapes(Torus_8, GeomAPI_Shape.FACE, [1])
53 model.testHaveNamingFaces(Torus_8, model, Part_1_doc)
54
55 model.testNbResults(Torus_11, 1)
56 model.testNbSubResults(Torus_11, [0])
57 model.testNbSubShapes(Torus_11, GeomAPI_Shape.SOLID, [1])
58 model.testNbSubShapes(Torus_11, GeomAPI_Shape.FACE, [1])
59 model.testHaveNamingFaces(Torus_11, model, Part_1_doc)
60
61 model.testNbResults(Torus_2, 0)
62 assert(Torus_2.feature().error() == "Torus builder :: ring radius is greater than the radius.")
63
64 model.testNbResults(Torus_3, 0)
65 assert(Torus_3.feature().error() == "Torus builder :: radius is negative or null.")
66
67 model.testNbResults(Torus_4, 0)
68 assert(Torus_4.feature().error() == "Torus builder :: radius is negative or null.")
69
70 model.testNbResults(Torus_5, 0)
71 assert(Torus_5.feature().error() == "Torus builder :: ring radius is negative or null.")
72
73 model.testNbResults(Torus_6, 0)
74 assert(Torus_6.feature().error() == "Torus builder :: ring radius is negative or null.")
75
76 model.testNbResults(Torus_9, 0)
77 assert(Torus_9.feature().error() == "Attribute \"base_point\" is not initialized.")
78
79 model.testNbResults(Torus_10, 0)
80 assert(Torus_10.feature().error() == "Attribute \"axis\" is not initialized.")