TheAlgorithms-Python

1
"""Uses Pythagoras theorem to calculate the distance between two points in space."""
2

3
import math
4

5

6
class Point:
7
    def __init__(self, x, y, z):
8
        self.x = x
9
        self.y = y
10
        self.z = z
11

12
    def __repr__(self) -> str:
13
        return f"Point({self.x}, {self.y}, {self.z})"
14

15

16
def distance(a: Point, b: Point) -> float:
17
    """
18
    >>> point1 = Point(2, -1, 7)
19
    >>> point2 = Point(1, -3, 5)
20
    >>> print(f"Distance from {point1} to {point2} is {distance(point1, point2)}")
21
    Distance from Point(2, -1, 7) to Point(1, -3, 5) is 3.0
22
    """
23
    return math.sqrt(abs((b.x - a.x) ** 2 + (b.y - a.y) ** 2 + (b.z - a.z) ** 2))
24

25

26
if __name__ == "__main__":
27
    import doctest
28

29
    doctest.testmod()
30