1 

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

     3 

     4 assert(collatz(2)==2)

     5 

     6 assert(collatz(9)==20)

     7 

     8 assert(collatz(9000)==48)

     9