src/PythonAPI/Test/TestSketcherSetAngle.py

   1 # Copyright (C) 2014-2023  CEA, EDF
   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 import unittest
  21 from salome.shaper import model
  22 import math
  23 import TestSketcher
  24 from TestSketcher import SketcherTestCase
  25
  26 class SketcherSetAngle(SketcherTestCase):
  27     def runTest(self):
  28         # Set the constraint
  29         l1 = self.sketch.addLine(0, 0, 0, 1)
  30         l2 = self.sketch.addLine(0, 0, 1, 1)
  31         self.sketch.setAngle(l1.result(), l2.result(), 30.0)
  32         # Commit the transaction
  33         model.do()
  34         # Check the result
  35         dot_product = (l1.endPoint().x() - l1.startPoint().x()) * \
  36                       (l2.endPoint().x() - l2.startPoint().x()) + \
  37                       (l1.endPoint().y() - l1.startPoint().y()) * \
  38                       (l2.endPoint().y() - l2.startPoint().y())
  39         norm_1 = math.sqrt(
  40             math.pow((l1.endPoint().x() - l1.startPoint().x()), 2) +
  41             math.pow((l1.endPoint().y() - l1.startPoint().y()), 2)
  42             )
  43         norm_2 = math.sqrt(
  44             math.pow((l2.endPoint().x() - l2.startPoint().x()), 2) +
  45             math.pow((l2.endPoint().y() - l2.startPoint().y()), 2)
  46             )
  47         angle = math.acos(dot_product / (norm_1 * norm_2))
  48         self.assertAlmostEqual(
  49             angle * 180 / math.pi, 30.0, delta=TestSketcher.DELTA
  50             )
  51
  52 if __name__ == "__main__":
  53     test_program = unittest.main(exit=False)
  54     assert test_program.result.wasSuccessful(), "Test failed"