Salome HOME
Add copyright header according to request of CEA from 06.06.2017
[modules/shaper.git] / src / PythonAPI / Test / TestPrimitivesBox.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 import unittest
22
23 import ModelAPI
24
25 from salome.shaper import model
26
27 class PrimitivesAddBox(unittest.TestCase):
28
29     def setUp(self):
30         model.begin()
31         # Create part
32         partset = model.moduleDocument()
33         self.part = model.addPart(partset).document()
34         model.do()
35
36     def tearDown(self):
37         model.end()
38         model.reset()
39
40 #-----------------------------------------------------------------------------
41 # TestCases
42
43 class PrimitivesAddBoxTestCase(PrimitivesAddBox):
44
45     def test_add_box_by_dimensions(self):
46         box = model.addBox(self.part, 50, 20, 10)
47         model.do()
48         self.assertEqual(box.creationMethod().value(),"BoxByDimensions")
49         self.assertEqual(box.dx().value(),50)
50         self.assertEqual(box.dy().value(),20)
51         self.assertEqual(box.dz().value(),10)
52
53     def test_add_box_by_two_points(self):
54         point1 = model.addPoint(self.part,0,0,0).result()
55         point2 = model.addPoint(self.part,10,10,10).result()
56         box = model.addBox(self.part, point1, point2)
57         model.do()
58         self.assertEqual(box.creationMethod().value(),"BoxByTwoPoints")
59         self.assertEqual(box.firstPoint().context().shape().isVertex(),True)
60         self.assertEqual(box.secondPoint().context().shape().isVertex(),True)
61
62 if __name__ == "__main__":
63     unittest.main()