public/go/tests/collatzcountdown/main.go

package main

import (
	"./student"
	"github.com/01-edu/public/go/lib"
)

func collatzCountdown(start int) int {
	if start <= 0 {
		return -1
	}

	steps := 0

	for start != 1 {
		if start%2 == 0 {
			start = start / 2
		} else {
			start = 3*start + 1
		}
		steps++
	}

	return steps
}

func main() {
	args := []int{lib.RandIntBetween(-6, 20)}
	args = append(args, -5, 0)
	for i := 0; i < 20; i++ {
		args = append(args, lib.RandIntBetween(2, 20))
	}

	for _, v := range args {
		lib.Challenge("CollatzCountdown", student.CollatzCountdown, collatzCountdown, v)
	}
}