smesh/src/Tools/blocFissure/gmu/distance2.py
2014-01-09 14:20:44 +00:00

36 lines
992 B
Python

# -*- coding: utf-8 -*-
import logging
# -----------------------------------------------------------------------------
# --- calcul de distance carree entre deux points, jeux de coordonnees x,y,z
def distance2(xyz1, xyz2):
"""
carré de la distance entre deux points donnés par des triplets [x,y,z]
"""
#logging.info("start")
d2 = 0
for i in range(3):
d2 += (xyz1[i]-xyz2[i])*(xyz1[i]-xyz2[i])
logging.debug('d2=%s', d2)
return d2
# -----------------------------------------------------------------------------
# --- test unitaire
import unittest
class Test_distance2(unittest.TestCase):
def setUp(self):
self.a=[0, 0, 0]
self.b=[3, 4, 5]
self.c=[-5,-4,-3]
def test_calcul(self):
self.assertEqual(distance2(self.a, self.b), distance2(self.b, self.a))
self.assertEqual(distance2(self.a, self.b), distance2(self.a, self.c))
self.assertEqual(distance2(self.b, self.b), 0)
self.assertEqual(distance2(self.a, self.b), 50)