ByoungSooPark/cadquery
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use gp_Vec.IsEqual() and add a test for Vector.__eq__()

Browse Source
This commit is contained in:
Justin Buchanan
2018-11-18 13:02:21 -08:00
parent 21388385fa
commit 084db38d56
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cadquery/occ_impl/geom.py
View File

@ -133,7 +133,7 @@ class Vector(object):
return 'Vector: ' + str((self.x, self.y, self.z)) return 'Vector: ' + str((self.x, self.y, self.z))
def __eq__(self, other): def __eq__(self, other):
return self.x == other.x and self.y == other.y and self.z == other.z return self.wrapped.IsEqual(other.wrapped, 0.00001, 0.00001)
''' '''
is not implemented in OCC is not implemented in OCC
def __ne__(self, other): def __ne__(self, other):

7
tests/TestCadObjects.py
View File

@ -108,6 +108,13 @@ class TestCadObjects(BaseTest):
result = Vector(1, 2, 0) + Vector(0, 0, 3) result = Vector(1, 2, 0) + Vector(0, 0, 3)
self.assertTupleAlmostEquals((1.0, 2.0, 3.0), result.toTuple(), 3) self.assertTupleAlmostEquals((1.0, 2.0, 3.0), result.toTuple(), 3)
def testVectorEquals(self):
a = Vector(1, 2, 3)
b = Vector(1, 2, 3)
c = Vector(1, 2, 3.000001)
self.assertEqual(a, b)
self.assertEqual(a, c)
def testTranslate(self): def testTranslate(self):
e = Edge.makeCircle(2, (1, 2, 3)) e = Edge.makeCircle(2, (1, 2, 3))
e2 = e.translate(Vector(0, 0, 1)) e2 = e.translate(Vector(0, 0, 1))