FilesExpand file tree

 Write-for-current-Python

/

pythagoras.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

33 lines (22 loc) · 752 Bytes

 Write-for-current-Python

/

pythagoras.py

Copy path

Top

File metadata and controls

• Code
• Blame

33 lines (22 loc) · 752 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

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

import math

class Point:

def __init__(self, x, y, z):

self.x = x

self.y = y

self.z = z

def __repr__(self) -> str:

return f"Point({self.x}, {self.y}, {self.z})"

def distance(a: Point, b: Point) -> float:

return math.sqrt(abs((b.x - a.x) ** 2 + (b.y - a.y) ** 2 + (b.z - a.z) ** 2))

def test_distance() -> None:

"""

>>> point1 = Point(2, -1, 7)

>>> point2 = Point(1, -3, 5)

>>> print(f"Distance from {point1} to {point2} is {distance(point1, point2)}")

Distance from Point(2, -1, 7) to Point(1, -3, 5) is 3.0

"""

pass

if __name__ == "__main__":

import doctest

doctest.testmod()