package main import ( "strconv" "strings" "testing" "github.com/01-edu/z01" ) func isAPrime(nb int) bool { if nb <= 0 || nb == 1 { return false } if nb <= 3 { return true } else if nb%2 == 0 || nb%3 == 0 { return false } i := 5 for i*i <= nb { if nb%i == 0 || nb%(i+2) == 0 { return false } i = i + 6 } return true } func TestAddPrimeSum(t *testing.T) { var table []string // fill with all rpime numbers between 0 and 100 for i := 0; i < 100; i++ { if isAPrime(i) { str := strconv.Itoa(i) table = append(table, str) } } // adds 15 random numbers for i := 0; i < 15; i++ { table = append(table, strconv.Itoa(z01.RandIntBetween(1, 10000))) } // special cases table = append(table, "\"\"") table = append(table, "1 2") for _, s := range table { z01.ChallengeMainExam(t, strings.Fields(s)...) } }