diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-16 19:25:22 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-16 19:25:22 +0000 |
commit | f6ad4dcef54c5ce997a4bad5a6d86de229015700 (patch) | |
tree | 7cfa4e31ace5c2bd95c72b154d15af494b2bcbef /test/map.go | |
parent | Initial commit. (diff) | |
download | golang-1.22-f6ad4dcef54c5ce997a4bad5a6d86de229015700.tar.xz golang-1.22-f6ad4dcef54c5ce997a4bad5a6d86de229015700.zip |
Adding upstream version 1.22.1.upstream/1.22.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'test/map.go')
-rw-r--r-- | test/map.go | 684 |
1 files changed, 684 insertions, 0 deletions
diff --git a/test/map.go b/test/map.go new file mode 100644 index 0000000..2c1cf8a --- /dev/null +++ b/test/map.go @@ -0,0 +1,684 @@ +// run + +// Copyright 2009 The Go Authors. All rights reserved. +// Use of this source code is governed by a BSD-style +// license that can be found in the LICENSE file. + +// Test maps, almost exhaustively. +// Complexity (linearity) test is in maplinear.go. + +package main + +import ( + "fmt" + "math" + "strconv" +) + +const count = 100 + +func P(a []string) string { + s := "{" + for i := 0; i < len(a); i++ { + if i > 0 { + s += "," + } + s += `"` + a[i] + `"` + } + s += "}" + return s +} + +func main() { + testbasic() + testfloat() + testnan() +} + +func testbasic() { + // Test a map literal. + mlit := map[string]int{"0": 0, "1": 1, "2": 2, "3": 3, "4": 4} + for i := 0; i < len(mlit); i++ { + s := string([]byte{byte(i) + '0'}) + if mlit[s] != i { + panic(fmt.Sprintf("mlit[%s] = %d\n", s, mlit[s])) + } + } + + mib := make(map[int]bool) + mii := make(map[int]int) + mfi := make(map[float32]int) + mif := make(map[int]float32) + msi := make(map[string]int) + mis := make(map[int]string) + mss := make(map[string]string) + mspa := make(map[string][]string) + // BUG need an interface map both ways too + + type T struct { + i int64 // can't use string here; struct values are only compared at the top level + f float32 + } + mipT := make(map[int]*T) + mpTi := make(map[*T]int) + mit := make(map[int]T) + // mti := make(map[T] int) + + type M map[int]int + mipM := make(map[int]M) + + var apT [2 * count]*T + + for i := 0; i < count; i++ { + s := strconv.Itoa(i) + s10 := strconv.Itoa(i * 10) + f := float32(i) + t := T{int64(i), f} + apT[i] = new(T) + apT[i].i = int64(i) + apT[i].f = f + apT[2*i] = new(T) // need twice as many entries as we use, for the nonexistence check + apT[2*i].i = int64(i) + apT[2*i].f = f + m := M{i: i + 1} + mib[i] = (i != 0) + mii[i] = 10 * i + mfi[float32(i)] = 10 * i + mif[i] = 10.0 * f + mis[i] = s + msi[s] = i + mss[s] = s10 + mss[s] = s10 + as := make([]string, 2) + as[0] = s10 + as[1] = s10 + mspa[s] = as + mipT[i] = apT[i] + mpTi[apT[i]] = i + mipM[i] = m + mit[i] = t + // mti[t] = i + } + + // test len + if len(mib) != count { + panic(fmt.Sprintf("len(mib) = %d\n", len(mib))) + } + if len(mii) != count { + panic(fmt.Sprintf("len(mii) = %d\n", len(mii))) + } + if len(mfi) != count { + panic(fmt.Sprintf("len(mfi) = %d\n", len(mfi))) + } + if len(mif) != count { + panic(fmt.Sprintf("len(mif) = %d\n", len(mif))) + } + if len(msi) != count { + panic(fmt.Sprintf("len(msi) = %d\n", len(msi))) + } + if len(mis) != count { + panic(fmt.Sprintf("len(mis) = %d\n", len(mis))) + } + if len(mss) != count { + panic(fmt.Sprintf("len(mss) = %d\n", len(mss))) + } + if len(mspa) != count { + panic(fmt.Sprintf("len(mspa) = %d\n", len(mspa))) + } + if len(mipT) != count { + panic(fmt.Sprintf("len(mipT) = %d\n", len(mipT))) + } + if len(mpTi) != count { + panic(fmt.Sprintf("len(mpTi) = %d\n", len(mpTi))) + } + // if len(mti) != count { + // panic(fmt.Sprintf("len(mti) = %d\n", len(mti))) + // } + if len(mipM) != count { + panic(fmt.Sprintf("len(mipM) = %d\n", len(mipM))) + } + // if len(mti) != count { + // panic(fmt.Sprintf("len(mti) = %d\n", len(mti))) + // } + if len(mit) != count { + panic(fmt.Sprintf("len(mit) = %d\n", len(mit))) + } + + // test construction directly + for i := 0; i < count; i++ { + s := strconv.Itoa(i) + s10 := strconv.Itoa(i * 10) + f := float32(i) + // BUG m := M(i, i+1) + if mib[i] != (i != 0) { + panic(fmt.Sprintf("mib[%d] = %t\n", i, mib[i])) + } + if mii[i] != 10*i { + panic(fmt.Sprintf("mii[%d] = %d\n", i, mii[i])) + } + if mfi[f] != 10*i { + panic(fmt.Sprintf("mfi[%d] = %d\n", i, mfi[f])) + } + if mif[i] != 10.0*f { + panic(fmt.Sprintf("mif[%d] = %g\n", i, mif[i])) + } + if mis[i] != s { + panic(fmt.Sprintf("mis[%d] = %s\n", i, mis[i])) + } + if msi[s] != i { + panic(fmt.Sprintf("msi[%s] = %d\n", s, msi[s])) + } + if mss[s] != s10 { + panic(fmt.Sprintf("mss[%s] = %g\n", s, mss[s])) + } + for j := 0; j < len(mspa[s]); j++ { + if mspa[s][j] != s10 { + panic(fmt.Sprintf("mspa[%s][%d] = %s\n", s, j, mspa[s][j])) + } + } + if mipT[i].i != int64(i) || mipT[i].f != f { + panic(fmt.Sprintf("mipT[%d] = %v\n", i, mipT[i])) + } + if mpTi[apT[i]] != i { + panic(fmt.Sprintf("mpTi[apT[%d]] = %d\n", i, mpTi[apT[i]])) + } + // if(mti[t] != i) { + // panic(fmt.Sprintf("mti[%s] = %s\n", s, mti[t])) + // } + if mipM[i][i] != i+1 { + panic(fmt.Sprintf("mipM[%d][%d] = %d\n", i, i, mipM[i][i])) + } + // if(mti[t] != i) { + // panic(fmt.Sprintf("mti[%v] = %d\n", t, mti[t])) + // } + if mit[i].i != int64(i) || mit[i].f != f { + panic(fmt.Sprintf("mit[%d] = {%d %g}\n", i, mit[i].i, mit[i].f)) + } + } + + // test existence with tuple check + // failed lookups yield a false value for the boolean. + for i := 0; i < count; i++ { + s := strconv.Itoa(i) + f := float32(i) + { + _, b := mib[i] + if !b { + panic(fmt.Sprintf("tuple existence decl: mib[%d]\n", i)) + } + _, b = mib[i] + if !b { + panic(fmt.Sprintf("tuple existence assign: mib[%d]\n", i)) + } + } + { + _, b := mii[i] + if !b { + panic(fmt.Sprintf("tuple existence decl: mii[%d]\n", i)) + } + _, b = mii[i] + if !b { + panic(fmt.Sprintf("tuple existence assign: mii[%d]\n", i)) + } + } + { + _, b := mfi[f] + if !b { + panic(fmt.Sprintf("tuple existence decl: mfi[%d]\n", i)) + } + _, b = mfi[f] + if !b { + panic(fmt.Sprintf("tuple existence assign: mfi[%d]\n", i)) + } + } + { + _, b := mif[i] + if !b { + panic(fmt.Sprintf("tuple existence decl: mif[%d]\n", i)) + } + _, b = mif[i] + if !b { + panic(fmt.Sprintf("tuple existence assign: mif[%d]\n", i)) + } + } + { + _, b := mis[i] + if !b { + panic(fmt.Sprintf("tuple existence decl: mis[%d]\n", i)) + } + _, b = mis[i] + if !b { + panic(fmt.Sprintf("tuple existence assign: mis[%d]\n", i)) + } + } + { + _, b := msi[s] + if !b { + panic(fmt.Sprintf("tuple existence decl: msi[%d]\n", i)) + } + _, b = msi[s] + if !b { + panic(fmt.Sprintf("tuple existence assign: msi[%d]\n", i)) + } + } + { + _, b := mss[s] + if !b { + panic(fmt.Sprintf("tuple existence decl: mss[%d]\n", i)) + } + _, b = mss[s] + if !b { + panic(fmt.Sprintf("tuple existence assign: mss[%d]\n", i)) + } + } + { + _, b := mspa[s] + if !b { + panic(fmt.Sprintf("tuple existence decl: mspa[%d]\n", i)) + } + _, b = mspa[s] + if !b { + panic(fmt.Sprintf("tuple existence assign: mspa[%d]\n", i)) + } + } + { + _, b := mipT[i] + if !b { + panic(fmt.Sprintf("tuple existence decl: mipT[%d]\n", i)) + } + _, b = mipT[i] + if !b { + panic(fmt.Sprintf("tuple existence assign: mipT[%d]\n", i)) + } + } + { + _, b := mpTi[apT[i]] + if !b { + panic(fmt.Sprintf("tuple existence decl: mpTi[apT[%d]]\n", i)) + } + _, b = mpTi[apT[i]] + if !b { + panic(fmt.Sprintf("tuple existence assign: mpTi[apT[%d]]\n", i)) + } + } + { + _, b := mipM[i] + if !b { + panic(fmt.Sprintf("tuple existence decl: mipM[%d]\n", i)) + } + _, b = mipM[i] + if !b { + panic(fmt.Sprintf("tuple existence assign: mipM[%d]\n", i)) + } + } + { + _, b := mit[i] + if !b { + panic(fmt.Sprintf("tuple existence decl: mit[%d]\n", i)) + } + _, b = mit[i] + if !b { + panic(fmt.Sprintf("tuple existence assign: mit[%d]\n", i)) + } + } + // { + // _, b := mti[t] + // if !b { + // panic(fmt.Sprintf("tuple existence decl: mti[%d]\n", i)) + // } + // _, b = mti[t] + // if !b { + // panic(fmt.Sprintf("tuple existence assign: mti[%d]\n", i)) + // } + // } + } + + // test nonexistence with tuple check + // failed lookups yield a false value for the boolean. + for i := count; i < 2*count; i++ { + s := strconv.Itoa(i) + f := float32(i) + { + _, b := mib[i] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: mib[%d]", i)) + } + _, b = mib[i] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: mib[%d]", i)) + } + } + { + _, b := mii[i] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: mii[%d]", i)) + } + _, b = mii[i] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: mii[%d]", i)) + } + } + { + _, b := mfi[f] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: mfi[%d]", i)) + } + _, b = mfi[f] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: mfi[%d]", i)) + } + } + { + _, b := mif[i] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: mif[%d]", i)) + } + _, b = mif[i] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: mif[%d]", i)) + } + } + { + _, b := mis[i] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: mis[%d]", i)) + } + _, b = mis[i] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: mis[%d]", i)) + } + } + { + _, b := msi[s] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: msi[%d]", i)) + } + _, b = msi[s] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: msi[%d]", i)) + } + } + { + _, b := mss[s] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: mss[%d]", i)) + } + _, b = mss[s] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: mss[%d]", i)) + } + } + { + _, b := mspa[s] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: mspa[%d]", i)) + } + _, b = mspa[s] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: mspa[%d]", i)) + } + } + { + _, b := mipT[i] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: mipT[%d]", i)) + } + _, b = mipT[i] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: mipT[%d]", i)) + } + } + { + _, b := mpTi[apT[i]] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: mpTi[apt[%d]]", i)) + } + _, b = mpTi[apT[i]] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: mpTi[apT[%d]]", i)) + } + } + { + _, b := mipM[i] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: mipM[%d]", i)) + } + _, b = mipM[i] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: mipM[%d]", i)) + } + } + // { + // _, b := mti[t] + // if b { + // panic(fmt.Sprintf("tuple nonexistence decl: mti[%d]", i)) + // } + // _, b = mti[t] + // if b { + // panic(fmt.Sprintf("tuple nonexistence assign: mti[%d]", i)) + // } + // } + { + _, b := mit[i] + if b { + panic(fmt.Sprintf("tuple nonexistence decl: mit[%d]", i)) + } + _, b = mit[i] + if b { + panic(fmt.Sprintf("tuple nonexistence assign: mit[%d]", i)) + } + } + } + + // tests for structured map element updates + for i := 0; i < count; i++ { + s := strconv.Itoa(i) + mspa[s][i%2] = "deleted" + if mspa[s][i%2] != "deleted" { + panic(fmt.Sprintf("update mspa[%s][%d] = %s\n", s, i%2, mspa[s][i%2])) + + } + + mipT[i].i += 1 + if mipT[i].i != int64(i)+1 { + panic(fmt.Sprintf("update mipT[%d].i = %d\n", i, mipT[i].i)) + + } + mipT[i].f = float32(i + 1) + if mipT[i].f != float32(i+1) { + panic(fmt.Sprintf("update mipT[%d].f = %g\n", i, mipT[i].f)) + + } + + mipM[i][i]++ + if mipM[i][i] != (i+1)+1 { + panic(fmt.Sprintf("update mipM[%d][%d] = %d\n", i, i, mipM[i][i])) + + } + } + + // test range on nil map + var mnil map[string]int + for _, _ = range mnil { + panic("range mnil") + } +} + +func testfloat() { + // Test floating point numbers in maps. + // Two map keys refer to the same entry if the keys are ==. + // The special cases, then, are that +0 == -0 and that NaN != NaN. + + { + var ( + pz = float32(0) + nz = math.Float32frombits(1 << 31) + nana = float32(math.NaN()) + nanb = math.Float32frombits(math.Float32bits(nana) ^ 2) + ) + + m := map[float32]string{ + pz: "+0", + nana: "NaN", + nanb: "NaN", + } + if m[pz] != "+0" { + panic(fmt.Sprintln("float32 map cannot read back m[+0]:", m[pz])) + } + if m[nz] != "+0" { + fmt.Sprintln("float32 map does not treat", pz, "and", nz, "as equal for read") + panic(fmt.Sprintln("float32 map does not treat -0 and +0 as equal for read")) + } + m[nz] = "-0" + if m[pz] != "-0" { + panic(fmt.Sprintln("float32 map does not treat -0 and +0 as equal for write")) + } + if _, ok := m[nana]; ok { + panic(fmt.Sprintln("float32 map allows NaN lookup (a)")) + } + if _, ok := m[nanb]; ok { + panic(fmt.Sprintln("float32 map allows NaN lookup (b)")) + } + if len(m) != 3 { + panic(fmt.Sprintln("float32 map should have 3 entries:", m)) + } + m[nana] = "NaN" + m[nanb] = "NaN" + if len(m) != 5 { + panic(fmt.Sprintln("float32 map should have 5 entries:", m)) + } + } + + { + var ( + pz = float64(0) + nz = math.Float64frombits(1 << 63) + nana = float64(math.NaN()) + nanb = math.Float64frombits(math.Float64bits(nana) ^ 2) + ) + + m := map[float64]string{ + pz: "+0", + nana: "NaN", + nanb: "NaN", + } + if m[nz] != "+0" { + panic(fmt.Sprintln("float64 map does not treat -0 and +0 as equal for read")) + } + m[nz] = "-0" + if m[pz] != "-0" { + panic(fmt.Sprintln("float64 map does not treat -0 and +0 as equal for write")) + } + if _, ok := m[nana]; ok { + panic(fmt.Sprintln("float64 map allows NaN lookup (a)")) + } + if _, ok := m[nanb]; ok { + panic(fmt.Sprintln("float64 map allows NaN lookup (b)")) + } + if len(m) != 3 { + panic(fmt.Sprintln("float64 map should have 3 entries:", m)) + } + m[nana] = "NaN" + m[nanb] = "NaN" + if len(m) != 5 { + panic(fmt.Sprintln("float64 map should have 5 entries:", m)) + } + } + + { + var ( + pz = complex64(0) + nz = complex(0, math.Float32frombits(1<<31)) + nana = complex(5, float32(math.NaN())) + nanb = complex(5, math.Float32frombits(math.Float32bits(float32(math.NaN()))^2)) + ) + + m := map[complex64]string{ + pz: "+0", + nana: "NaN", + nanb: "NaN", + } + if m[nz] != "+0" { + panic(fmt.Sprintln("complex64 map does not treat -0 and +0 as equal for read")) + } + m[nz] = "-0" + if m[pz] != "-0" { + panic(fmt.Sprintln("complex64 map does not treat -0 and +0 as equal for write")) + } + if _, ok := m[nana]; ok { + panic(fmt.Sprintln("complex64 map allows NaN lookup (a)")) + } + if _, ok := m[nanb]; ok { + panic(fmt.Sprintln("complex64 map allows NaN lookup (b)")) + } + if len(m) != 3 { + panic(fmt.Sprintln("complex64 map should have 3 entries:", m)) + } + m[nana] = "NaN" + m[nanb] = "NaN" + if len(m) != 5 { + panic(fmt.Sprintln("complex64 map should have 5 entries:", m)) + } + } + + { + var ( + pz = complex128(0) + nz = complex(0, math.Float64frombits(1<<63)) + nana = complex(5, float64(math.NaN())) + nanb = complex(5, math.Float64frombits(math.Float64bits(float64(math.NaN()))^2)) + ) + + m := map[complex128]string{ + pz: "+0", + nana: "NaN", + nanb: "NaN", + } + if m[nz] != "+0" { + panic(fmt.Sprintln("complex128 map does not treat -0 and +0 as equal for read")) + } + m[nz] = "-0" + if m[pz] != "-0" { + panic(fmt.Sprintln("complex128 map does not treat -0 and +0 as equal for write")) + } + if _, ok := m[nana]; ok { + panic(fmt.Sprintln("complex128 map allows NaN lookup (a)")) + } + if _, ok := m[nanb]; ok { + panic(fmt.Sprintln("complex128 map allows NaN lookup (b)")) + } + if len(m) != 3 { + panic(fmt.Sprintln("complex128 map should have 3 entries:", m)) + } + m[nana] = "NaN" + m[nanb] = "NaN" + if len(m) != 5 { + panic(fmt.Sprintln("complex128 map should have 5 entries:", m)) + } + } +} + +func testnan() { + n := 500 + m := map[float64]int{} + nan := math.NaN() + for i := 0; i < n; i++ { + m[nan] = 1 + } + if len(m) != n { + panic("wrong size map after nan insertion") + } + iters := 0 + for k, v := range m { + iters++ + if !math.IsNaN(k) { + panic("not NaN") + } + if v != 1 { + panic("wrong value") + } + } + if iters != n { + panic("wrong number of nan range iters") + } +} |