diff --git a/code_kata_230820/prime_factors_test.go b/code_kata_230820/prime_factors_test.go new file mode 100644 index 0000000..ca0be97 --- /dev/null +++ b/code_kata_230820/prime_factors_test.go @@ -0,0 +1,47 @@ +package main + +import ( + "reflect" + "testing" +) + +func PrimeFactorsOf(n int) []int { + ret := make([]int, 0) + d := 2 + for d < n { + for n%d == 0 { + ret = append(ret, d) + n /= d + } + d++ + } + if n > 1 { + ret = append(ret, n) + } + return ret +} + +func TestPrimeFactorsOf(t *testing.T) { + tt := []struct { + name string + input int + want []int + }{ + {"1", 1, []int{}}, + {"2", 2, []int{2}}, + {"3", 3, []int{3}}, + {"4", 4, []int{2, 2}}, + {"8", 8, []int{2, 2, 2}}, + {"9", 9, []int{3, 3}}, + {"a very large number", 2 * 3 * 7 * 17 * 37 * 71 * 73, []int{2, 3, 7, 17, 37, 71, 73}}, + } + + for _, tc := range tt { + t.Run(tc.name, func(t *testing.T) { + got := PrimeFactorsOf(tc.input) + if !reflect.DeepEqual(got, tc.want) { + t.Errorf("PrimeFactorsOf(%d) = %v, want %v\n", tc.input, got, tc.want) + } + }) + } +}