src/SketchPlugin/Test/TestConstraintCollinear.py

   1 """
   2     TestConstraintCollinear.py
   3     Unit test of SketchPlugin_ConstraintCollinear class
   4
   5     SketchPlugin_ConstraintCollinear
   6         static const std::string MY_CONSTRAINT_COLLINEAR_ID("SketchConstraintCollinear");
   7         data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::typeId());
   8         data()->addAttribute(SketchPlugin_Constraint::ENTITY_B(), ModelAPI_AttributeRefAttr::typeId());
   9
  10 """
  11
  12 #=========================================================================
  13 # Initialization of the test
  14 #=========================================================================
  15
  16 from GeomDataAPI import *
  17 from ModelAPI import *
  18 import math
  19 import unittest
  20 from salome.shaper import model
  21
  22 __updated__ = "2017-03-06"
  23
  24 class TestConstraintCollinear(unittest.TestCase):
  25   def setUp(self):
  26     model.begin()
  27     self.myDocument = model.moduleDocument()
  28     self.mySketch = model.addSketch(self.myDocument, model.defaultPlane("XOY"))
  29     self.myTolerance = 1.e-6
  30     self.myDOF = 0
  31
  32   def tearDown(self):
  33     model.end()
  34     assert(model.checkPythonDump())
  35
  36   def checkVectorCollinearity(self, theX1, theY1, theX2, theY2):
  37     aLen1 = math.hypot(theX1, theY1)
  38     aLen2 = math.hypot(theX2, theY2)
  39     aDot = theX1 * theX2 + theY1 * theY2
  40     self.assertTrue(math.fabs(math.fabs(aDot) - aLen1 * aLen2) < self.myTolerance**2, "Vectors ({0}, {1}) and ({2}, {3}) do not collinear".format(theX1, theY1, theX2, theY2))
  41
  42   def checkLineCollinearity(self, theLine1, theLine2):
  43     aStartPoint1 = theLine1.startPoint()
  44     aEndPoint1   = theLine1.endPoint()
  45     aStartPoint2 = theLine2.startPoint()
  46     aEndPoint2   = theLine2.endPoint()
  47
  48     aDir1x, aDir1y = aEndPoint1.x() - aStartPoint1.x(), aEndPoint1.y() - aStartPoint1.y()
  49     aDir2x, aDir2y = aEndPoint2.x() - aStartPoint1.x(), aEndPoint2.y() - aStartPoint1.y()
  50     aDir3x, aDir3y = aStartPoint2.x() - aStartPoint1.x(), aStartPoint2.y() - aStartPoint1.y()
  51     self.checkVectorCollinearity(aDir1x, aDir1y, aDir2x, aDir2y)
  52     self.checkVectorCollinearity(aDir1x, aDir1y, aDir3x, aDir3y)
  53
  54   def moveLineAndCheckCollinearity(self, theLine1, theLine2):
  55     deltaX = deltaY = 10.
  56
  57     theLine1.startPoint().setValue(theLine1.startPoint().x() + deltaX, theLine1.startPoint().y() + deltaY)
  58     model.do()
  59     self.checkLineCollinearity(theLine1, theLine2)
  60     self.assertEqual(model.dof(self.mySketch), self.myDOF)
  61
  62     theLine1.endPoint().setValue(theLine1.endPoint().x() - deltaX, theLine1.endPoint().y() - deltaY)
  63     model.do()
  64     self.checkLineCollinearity(theLine1, theLine2)
  65     self.assertEqual(model.dof(self.mySketch), self.myDOF)
  66
  67   def test_collinear_base(self):
  68     """ Test 1. Collinearity two independent lines
  69     """
  70     aSketchLineA = self.mySketch.addLine(0., 25., 85., 25.)
  71     aSketchLineB = self.mySketch.addLine(0., 50., 80., 75.)
  72     self.myDOF += 8
  73     self.assertEqual(model.dof(self.mySketch), self.myDOF)
  74     self.mySketch.setCollinear(aSketchLineA, aSketchLineB)
  75     self.myDOF -= 2
  76     model.do()
  77     self.assertEqual(model.dof(self.mySketch), self.myDOF)
  78
  79     self.checkLineCollinearity(aSketchLineA, aSketchLineB)
  80     self.moveLineAndCheckCollinearity(aSketchLineA, aSketchLineB)
  81
  82   def test_collinear_connected_lines(self):
  83     """ Test 2. Collinearity of two lines in polyline
  84     """
  85     aSketchLineA = self.mySketch.addLine(10., 20., 30., 40.)
  86     aSketchLineB = self.mySketch.addLine(30., 40., 30., 70.)
  87     self.myDOF += 8
  88     self.assertEqual(model.dof(self.mySketch), self.myDOF)
  89     self.mySketch.setCoincident(aSketchLineA.endPoint(), aSketchLineB.startPoint())
  90     self.myDOF -= 2
  91     model.do()
  92     self.assertEqual(model.dof(self.mySketch), self.myDOF)
  93
  94     self.mySketch.setCollinear(aSketchLineA, aSketchLineB)
  95     self.myDOF -= 1
  96     model.do()
  97     self.assertEqual(model.dof(self.mySketch), self.myDOF)
  98
  99     self.checkLineCollinearity(aSketchLineA, aSketchLineB)
 100     self.moveLineAndCheckCollinearity(aSketchLineA, aSketchLineB)
 101
 102   def test_collinear_point_on_line(self):
 103     """ Test 3. Collinearity for line which extremity is coincident with other line
 104     """
 105     aSketchLineA = self.mySketch.addLine(10., 20., 30., 40.)
 106     aSketchLineB = self.mySketch.addLine(20., 40., 30., 70.)
 107     self.myDOF += 8
 108     self.assertEqual(model.dof(self.mySketch), self.myDOF)
 109     self.mySketch.setCoincident(aSketchLineA.result(), aSketchLineB.startPoint())
 110     self.myDOF -= 1
 111     model.do()
 112     self.assertEqual(model.dof(self.mySketch), self.myDOF)
 113
 114     self.mySketch.setCollinear(aSketchLineA, aSketchLineB)
 115     self.myDOF -= 1
 116     model.do()
 117     self.assertEqual(model.dof(self.mySketch), self.myDOF)
 118
 119     self.checkLineCollinearity(aSketchLineA, aSketchLineB)
 120     self.moveLineAndCheckCollinearity(aSketchLineA, aSketchLineB)
 121
 122   def test_already_collinear(self):
 123     """ Test 4. Collinearity two lines connected by extremities to each other
 124     """
 125     aSketchLineA = self.mySketch.addLine(10., 20., 40., 50.)
 126     aSketchLineB = self.mySketch.addLine(20., 40., 40., 70.)
 127     self.myDOF += 8
 128     self.assertEqual(model.dof(self.mySketch), self.myDOF)
 129     self.mySketch.setCoincident(aSketchLineA.result(), aSketchLineB.startPoint())
 130     self.mySketch.setCoincident(aSketchLineA.endPoint(), aSketchLineB.result())
 131     self.myDOF -= 2
 132     model.do()
 133     self.assertEqual(model.dof(self.mySketch), self.myDOF)
 134
 135     aCollinear = self.mySketch.setCollinear(aSketchLineA, aSketchLineB)
 136     model.do()
 137     self.assertEqual(model.dof(self.mySketch), self.myDOF)
 138
 139     # check error message and remove non valid constraint
 140     self.assertNotEqual(self.mySketch.solverError().value(), "")
 141     self.myDocument.removeFeature(aCollinear.feature())
 142     model.do()
 143     self.assertEqual(model.dof(self.mySketch), self.myDOF)
 144
 145   def test_already_collinear2(self):
 146     """ Test 5. Collinearity of two lines when one line is fully on other
 147     """
 148     aSketchLineA = self.mySketch.addLine(10., 20., 30., 40.)
 149     aSketchLineB = self.mySketch.addLine(20., 40., 30., 70.)
 150     self.myDOF += 8
 151     self.assertEqual(model.dof(self.mySketch), self.myDOF)
 152     self.mySketch.setCoincident(aSketchLineA.result(), aSketchLineB.startPoint())
 153     self.mySketch.setCoincident(aSketchLineA.result(), aSketchLineB.endPoint())
 154     self.myDOF -= 2
 155     model.do()
 156     self.assertEqual(model.dof(self.mySketch), self.myDOF)
 157
 158     aCollinear = self.mySketch.setCollinear(aSketchLineA, aSketchLineB)
 159     model.do()
 160     self.assertEqual(model.dof(self.mySketch), self.myDOF)
 161
 162     # check error message and remove non valid constraint
 163     self.assertNotEqual(self.mySketch.solverError().value(), "")
 164     self.myDocument.removeFeature(aCollinear.feature())
 165     model.do()
 166     self.assertEqual(model.dof(self.mySketch), self.myDOF)
 167
 168 if __name__ == '__main__':
 169   unittest.main()
 170 #=========================================================================
 171 # End of test
 172 #=========================================================================