1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
package factors
import (
"fmt"
"testing"
)
var primeFactorCases = map[uint]PrimeFactorization{
0: PrimeFactorization{map[uint]uint{0: 1}},
1: PrimeFactorization{map[uint]uint{}},
2: PrimeFactorization{map[uint]uint{2: 1}},
3: PrimeFactorization{map[uint]uint{3: 1}},
4: PrimeFactorization{map[uint]uint{2: 2}},
6: PrimeFactorization{map[uint]uint{2: 1, 3: 1}},
10: PrimeFactorization{map[uint]uint{2: 1, 5: 1}},
12: PrimeFactorization{map[uint]uint{2: 2, 3: 1}},
33: PrimeFactorization{map[uint]uint{3: 1, 11: 1}},
60: PrimeFactorization{map[uint]uint{2: 2, 3: 1, 5: 1}},
86400: PrimeFactorization{map[uint]uint{2: 7, 3: 3, 5: 2}},
}
func TestPrimeFactorize(t *testing.T) {
for i, expected := range primeFactorCases {
testname := fmt.Sprintf("%d", i)
t.Run(testname, func(t *testing.T) {
actual := PrimeFactorize(i)
if !mapEquals(expected.exponents, actual.exponents) {
t.Errorf("PrimeFactorize(%d) = %s, want %s", i, actual, expected)
}
})
}
}
func mapEquals(a, b map[uint]uint) bool {
for k := range a {
if a[k] != b[k] {
return false
}
}
for k := range b {
if a[k] != b[k] {
return false
}
}
return true
}
|
