summaryrefslogtreecommitdiff
path: root/factors/table_test.go
diff options
context:
space:
mode:
authorAdrien Hopkins <adrien.p.hopkins@gmail.com>2023-11-07 07:50:23 -0500
committerAdrien Hopkins <adrien.p.hopkins@gmail.com>2023-11-09 19:47:16 -0500
commit2cc0f9607872b8bd4fc6be2656a7b7a769b538b2 (patch)
tree5e14b4ffcbf7c0d85f17b0ed468a08b4bafa11c1 /factors/table_test.go
parent7d2916a50187992f72e80ed667468e8c0b125d27 (diff)
Reduce golang requirement to go1.18
go1.21, the previous requirement, was released a few months ago, so not all systems have adopted it. go1.18 is old enough that most systems should support it, but it introduces generics, which my testing code is highly dependent on, so I can't easily go any earlier.
Diffstat (limited to 'factors/table_test.go')
-rw-r--r--factors/table_test.go66
1 files changed, 48 insertions, 18 deletions
diff --git a/factors/table_test.go b/factors/table_test.go
index e288c3b..828d80b 100644
--- a/factors/table_test.go
+++ b/factors/table_test.go
@@ -2,8 +2,6 @@ package factors
import (
"fmt"
- "maps"
- "slices"
"testing"
)
@@ -41,7 +39,7 @@ var primeFactorCases = map[uint]PrimeFactorization{
func TestPrimeFactorize(t *testing.T) {
equal := func(a, b PrimeFactorization) bool {
- return maps.Equal(a.exponents, b.exponents)
+ return mapEquals(a.exponents, b.exponents)
}
tableTest(t, PrimeFactorize, primeFactorCases, equal, "PrimeFactorize")
}
@@ -66,7 +64,7 @@ var factorCases = map[uint][]uint{
}
func TestFactors(t *testing.T) {
- tableTest(t, Factors, factorCases, setEquals, "Factors")
+ tableTest(t, Factors, factorCases, setEquals[uint], "Factors")
}
var totativeRatioCases = map[uint]float64{
@@ -81,7 +79,8 @@ func totativeRatio(n uint) float64 {
}
func TestTotativeRatio(t *testing.T) {
- tableTest(t, totativeRatio, totativeRatioCases, stdEquals, "TotativeRatio")
+ tableTest(t, totativeRatio, totativeRatioCases, stdEquals[float64],
+ "TotativeRatio")
}
var totientCases = map[uint]uint{
@@ -90,7 +89,7 @@ var totientCases = map[uint]uint{
}
func TestTotativeCount(t *testing.T) {
- tableTest(t, Totient, totientCases, stdEquals, "Totient")
+ tableTest(t, Totient, totientCases, stdEquals[uint], "Totient")
}
var totativeDigitCases = map[uint32][]uint32{
@@ -121,7 +120,7 @@ var totativeDigitCases = map[uint32][]uint32{
}
func TestTotativeDigits(t *testing.T) {
- tableTest(t, TotativeDigits, totativeDigitCases, slices.Equal,
+ tableTest(t, TotativeDigits, totativeDigitCases, sliceEquals[uint32],
"TotativeDigits")
}
@@ -142,7 +141,7 @@ var factorScoreCases = map[uint]float64{
func TestFactorScore(t *testing.T) {
// factors.Score is accurate enough that we can test for exact floats!
- tableTest(t, Score, factorScoreCases, stdEquals, "Score")
+ tableTest(t, Score, factorScoreCases, stdEquals[float64], "Score")
}
var basicRankCases = map[uint]string{
@@ -160,7 +159,7 @@ var basicRankCases = map[uint]string{
}
func TestBasicRank(t *testing.T) {
- tableTest(t, BasicRank, basicRankCases, stdEquals, "BasicRank")
+ tableTest(t, BasicRank, basicRankCases, stdEquals[string], "BasicRank")
}
var gcdCases = map[struct{ a, b uint32 }]uint32{
@@ -181,7 +180,7 @@ func gcdTest(c struct{ a, b uint32 }) uint32 {
}
func TestGCD(t *testing.T) {
- tableTest(t, gcdTest, gcdCases, stdEquals, "gcd")
+ tableTest(t, gcdTest, gcdCases, stdEquals[uint32], "gcd")
}
var mtcCases = map[uint32]uint64{
@@ -194,7 +193,7 @@ var mtcCases = map[uint32]uint64{
}
func TestMTC(t *testing.T) {
- tableTest(t, MTC, mtcCases, stdEquals, "MTC")
+ tableTest(t, MTC, mtcCases, stdEquals[uint64], "MTC")
}
var sanCases = map[uint]bool{
@@ -212,7 +211,7 @@ func isSAN(n uint) bool {
}
func TestSAN(t *testing.T) {
- tableTest(t, isSAN, sanCases, stdEquals, "isSAN")
+ tableTest(t, isSAN, sanCases, stdEquals[bool], "isSAN")
}
var expOrdCases = map[uint]bool{
@@ -223,7 +222,8 @@ var expOrdCases = map[uint]bool{
}
func TestExponentsOrdered(t *testing.T) {
- tableTest(t, exponentsOrdered, expOrdCases, stdEquals, "exponentsOrdered")
+ tableTest(t, exponentsOrdered, expOrdCases, stdEquals[bool],
+ "exponentsOrdered")
}
var practicalCases = map[uint]bool{
@@ -237,7 +237,7 @@ var practicalCases = map[uint]bool{
}
func TestPractical(t *testing.T) {
- tableTest(t, practical, practicalCases, stdEquals, "practical")
+ tableTest(t, practical, practicalCases, stdEquals[bool], "practical")
}
var typeCases = map[uint]CompositenessType{
@@ -255,7 +255,7 @@ var typeCases = map[uint]CompositenessType{
}
func TestType(t *testing.T) {
- tableTest(t, Type, typeCases, stdEquals, "Type")
+ tableTest(t, Type, typeCases, stdEquals[CompositenessType], "Type")
}
// ====== DIGIT MAP TESTS ======
@@ -302,7 +302,7 @@ var digitMapCases = map[uint][]DigitType{
}
func TestDigitMap(t *testing.T) {
- tableTest(t, DigitMap, digitMapCases, slices.Equal, "DigitMap")
+ tableTest(t, DigitMap, digitMapCases, sliceEquals[DigitType], "DigitMap")
}
// ensures GetDigitType(d, r) == DigitMap(r)[d];
@@ -383,12 +383,28 @@ func splitTest(a uintPair) uintPair {
}
func TestSplit(t *testing.T) {
- tableTest(t, splitTest, splitCases, stdEquals, "Split")
+ tableTest(t, splitTest, splitCases, stdEquals[uintPair], "Split")
}
// to be used as the equal paramater for tableTest
func stdEquals[T comparable](a, b T) bool { return a == b }
+func mapEquals[K, V comparable](a, b map[K]V) bool {
+ for k, av := range a {
+ bv, ok := b[k]
+ if !ok || bv != av {
+ return false
+ }
+ }
+ for k, bv := range b {
+ av, ok := a[k]
+ if !ok || bv != av {
+ return false
+ }
+ }
+ return true
+}
+
func setEquals[E comparable](a, b []E) bool {
// use maps to simulate sets
// aSet[a] == true means set contains a, false means not
@@ -400,5 +416,19 @@ func setEquals[E comparable](a, b []E) bool {
for _, j := range b {
bSet[j] = true
}
- return maps.Equal(aSet, bSet)
+ return mapEquals(aSet, bSet)
+}
+
+func sliceEquals[E comparable](a, b []E) bool {
+ if len(a) != len(b) {
+ return false
+ }
+
+ for i := 0; i < len(a); i++ {
+ if a[i] != b[i] {
+ return false
+ }
+ }
+
+ return true
}