TDT4109/Exercise 1/9 - Tetraeder.py

from math import sqrt

def inputTypeCheck(message, type, errorMessage):
  while True:
    inputValue = input(message)
    try:
      return type(inputValue)
    except ValueError:
      print(errorMessage)

class Tetraeder:
  def __init__(self, length):
    self.length = length
    self.a = 3/sqrt(6) * length

  getArea = lambda self: sqrt(3) * (self.a**2)
  getVolume = lambda self: sqrt(2) * (self.a**3) / 12

figure1 = Tetraeder(3)
print(f'Et tetraeder med høyde {figure1.length} har areal {figure1.getArea()}')
print(f'Et tetraeder med høyde {figure1.length} har volum {figure1.getVolume()}')
print()

figure2 = Tetraeder(
  inputTypeCheck(
    message='Skriv inn en høyde: ',
    type=int,
    errorMessage='Beklager, det du skrev inn er ikke et heltall. Prøv igjen\n'
  )
)

print(f'Et tetraeder med høyde {figure1.length} har volum {figure2.getVolume()} og areal {figure2.getArea()}')