Python implementation of the Collatz series as a generator

collatz.py

def next_collatz(n):
  if n%2 == 0:
    return n//2
  else:
    return 3*n+1
  
def collatz(n):
  yield n
  result = next_collatz(n)
  while result != 1 and result :
    yield result
    result = next_collatz(result)
  # when we arrive here, the final result is 1
  yield 1
  return