Collatz function

collatz.py

def collatz(n: int) -> int:
  return n // 2 if n % 2 == 0 else 3*n + 1

def height(n: int) -> int:
  # The Collatz height of an integer is the # of steps to return to 1
  h = 0
  while n > 1:
    n = collatz(n)
    h += 1
  return h
  
def height_r(n: int) -> int:
  # Recursive implementation of Collatz height
  return 1 + height_r(collatz(n)) if n > 1 else 0