1 

     2 assert(List(0,1,4).contains(collatz(1)))

     3 

     4 assert(collatz(6) == 9)

     5 

     6 assert(collatz(9) == 20)

     7 

     8