Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 1 | // Copyright 2017 The Bazel Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style |
| 3 | // license that can be found in the LICENSE file. |
| 4 | |
Alan Donovan | e3deafe | 2018-10-23 11:05:09 -0400 | [diff] [blame] | 5 | package starlark |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 6 | |
| 7 | import ( |
alandonovan | f94b021 | 2018-10-12 15:31:48 -0400 | [diff] [blame] | 8 | "fmt" |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 9 | "math/rand" |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 10 | "sync" |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 11 | "testing" |
| 12 | ) |
| 13 | |
| 14 | func TestHashtable(t *testing.T) { |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 15 | makeTestIntsOnce.Do(makeTestInts) |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 16 | testHashtable(t, make(map[int]bool)) |
| 17 | } |
| 18 | |
alandonovan | f94b021 | 2018-10-12 15:31:48 -0400 | [diff] [blame] | 19 | func BenchmarkStringHash(b *testing.B) { |
| 20 | for len := 1; len <= 1024; len *= 2 { |
| 21 | buf := make([]byte, len) |
| 22 | rand.New(rand.NewSource(0)).Read(buf) |
| 23 | s := string(buf) |
| 24 | |
| 25 | b.Run(fmt.Sprintf("hard-%d", len), func(b *testing.B) { |
| 26 | for i := 0; i < b.N; i++ { |
| 27 | hashString(s) |
| 28 | } |
| 29 | }) |
| 30 | b.Run(fmt.Sprintf("soft-%d", len), func(b *testing.B) { |
| 31 | for i := 0; i < b.N; i++ { |
| 32 | softHashString(s) |
| 33 | } |
| 34 | }) |
| 35 | } |
| 36 | } |
| 37 | |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 38 | func BenchmarkHashtable(b *testing.B) { |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 39 | makeTestIntsOnce.Do(makeTestInts) |
| 40 | b.ResetTimer() |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 41 | for i := 0; i < b.N; i++ { |
| 42 | testHashtable(b, nil) |
| 43 | } |
| 44 | } |
| 45 | |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 46 | const testIters = 10000 |
| 47 | |
| 48 | var ( |
| 49 | // testInts is a zipf-distributed array of Ints and corresponding ints. |
| 50 | // This removes the cost of generating them on the fly during benchmarking. |
| 51 | // Without this, Zipf and MakeInt dominate CPU and memory costs, respectively. |
| 52 | makeTestIntsOnce sync.Once |
| 53 | testInts [3 * testIters]struct { |
| 54 | Int Int |
| 55 | goInt int |
Josh Bleecher Snyder | 107fc20 | 2018-11-24 08:47:54 -0800 | [diff] [blame] | 56 | } |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 57 | ) |
| 58 | |
| 59 | func makeTestInts() { |
| 60 | zipf := rand.NewZipf(rand.New(rand.NewSource(0)), 1.1, 1.0, 1000.0) |
| 61 | for i := range &testInts { |
| 62 | r := int(zipf.Uint64()) |
| 63 | testInts[i].goInt = r |
| 64 | testInts[i].Int = MakeInt(r) |
| 65 | } |
Josh Bleecher Snyder | 107fc20 | 2018-11-24 08:47:54 -0800 | [diff] [blame] | 66 | } |
| 67 | |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 68 | // testHashtable is both a test and a benchmark of hashtable. |
| 69 | // When sane != nil, it acts as a test against the semantics of Go's map. |
| 70 | func testHashtable(tb testing.TB, sane map[int]bool) { |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 71 | var i int // index into testInts |
Josh Bleecher Snyder | 107fc20 | 2018-11-24 08:47:54 -0800 | [diff] [blame] | 72 | |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 73 | var ht hashtable |
| 74 | |
| 75 | // Insert 10000 random ints into the map. |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 76 | for j := 0; j < testIters; j++ { |
| 77 | k := testInts[i] |
Josh Bleecher Snyder | 107fc20 | 2018-11-24 08:47:54 -0800 | [diff] [blame] | 78 | i++ |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 79 | if err := ht.insert(k.Int, None); err != nil { |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 80 | tb.Fatal(err) |
| 81 | } |
| 82 | if sane != nil { |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 83 | sane[k.goInt] = true |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 84 | } |
| 85 | } |
| 86 | |
| 87 | // Do 10000 random lookups in the map. |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 88 | for j := 0; j < testIters; j++ { |
| 89 | k := testInts[i] |
Josh Bleecher Snyder | 107fc20 | 2018-11-24 08:47:54 -0800 | [diff] [blame] | 90 | i++ |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 91 | _, found, err := ht.lookup(k.Int) |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 92 | if err != nil { |
| 93 | tb.Fatal(err) |
| 94 | } |
| 95 | if sane != nil { |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 96 | _, found2 := sane[k.goInt] |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 97 | if found != found2 { |
| 98 | tb.Fatal("sanity check failed") |
| 99 | } |
| 100 | } |
| 101 | } |
| 102 | |
| 103 | // Do 10000 random deletes from the map. |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 104 | for j := 0; j < testIters; j++ { |
| 105 | k := testInts[i] |
Josh Bleecher Snyder | 107fc20 | 2018-11-24 08:47:54 -0800 | [diff] [blame] | 106 | i++ |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 107 | _, found, err := ht.delete(k.Int) |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 108 | if err != nil { |
| 109 | tb.Fatal(err) |
| 110 | } |
| 111 | if sane != nil { |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 112 | _, found2 := sane[k.goInt] |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 113 | if found != found2 { |
| 114 | tb.Fatal("sanity check failed") |
| 115 | } |
Josh Bleecher Snyder | 5eb2bff | 2018-11-30 14:04:33 -0800 | [diff] [blame] | 116 | delete(sane, k.goInt) |
Alan Donovan | 312d1a5 | 2017-10-02 10:10:28 -0400 | [diff] [blame] | 117 | } |
| 118 | } |
| 119 | |
| 120 | if sane != nil { |
| 121 | if int(ht.len) != len(sane) { |
| 122 | tb.Fatal("sanity check failed") |
| 123 | } |
| 124 | } |
| 125 | } |