src/PythonAPI/Test/TestPrimitivesBox.py

   1 import unittest
   2
   3 import ModelAPI
   4
   5 from salome.shaper import model
   6
   7 class PrimitivesAddBox(unittest.TestCase):
   8
   9     def setUp(self):
  10         model.begin()
  11         # Create part
  12         partset = model.moduleDocument()
  13         self.part = model.addPart(partset).document()
  14         model.do()
  15
  16     def tearDown(self):
  17         model.end()
  18         model.reset()
  19
  20 #-----------------------------------------------------------------------------
  21 # TestCases
  22
  23 class PrimitivesAddBoxTestCase(PrimitivesAddBox):
  24
  25     def test_add_box_by_dimensions(self):
  26         box = model.addBox(self.part, 50, 20, 10)
  27         model.do()
  28         self.assertEqual(box.creationMethod().value(),"BoxByDimensions")
  29         self.assertEqual(box.dx().value(),50)
  30         self.assertEqual(box.dy().value(),20)
  31         self.assertEqual(box.dz().value(),10)
  32
  33     def test_add_box_by_two_points(self):
  34         point1 = model.addPoint(self.part,0,0,0).result()
  35         point2 = model.addPoint(self.part,10,10,10).result()
  36         box = model.addBox(self.part, point1, point2)
  37         model.do()
  38         self.assertEqual(box.creationMethod().value(),"BoxByTwoPoints")
  39         self.assertEqual(box.firstPoint().context().shape().isVertex(),True)
  40         self.assertEqual(box.secondPoint().context().shape().isVertex(),True)
  41
  42 if __name__ == "__main__":
  43     unittest.main()