runtime: copy mstats code from Go 1.7 runtime

[official-gcc.git] / libgo / go / sort / sort_test.go

// 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.

package sort_test

import (
        "fmt"
        "math"
        "math/rand"
        . "sort"
        "strconv"
        "testing"
)

var ints = [...]int{74, 59, 238, -784, 9845, 959, 905, 0, 0, 42, 7586, -5467984, 7586}
var float64s = [...]float64{74.3, 59.0, math.Inf(1), 238.2, -784.0, 2.3, math.NaN(), math.NaN(), math.Inf(-1), 9845.768, -959.7485, 905, 7.8, 7.8}
var strings = [...]string{"", "Hello", "foo", "bar", "foo", "f00", "%*&^*&^&", "***"}

func TestSortIntSlice(t *testing.T) {
        data := ints
        a := IntSlice(data[0:])
        Sort(a)
        if !IsSorted(a) {
                t.Errorf("sorted %v", ints)
                t.Errorf("   got %v", data)
        }
}

func TestSortFloat64Slice(t *testing.T) {
        data := float64s
        a := Float64Slice(data[0:])
        Sort(a)
        if !IsSorted(a) {
                t.Errorf("sorted %v", float64s)
                t.Errorf("   got %v", data)
        }
}

func TestSortStringSlice(t *testing.T) {
        data := strings
        a := StringSlice(data[0:])
        Sort(a)
        if !IsSorted(a) {
                t.Errorf("sorted %v", strings)
                t.Errorf("   got %v", data)
        }
}

func TestInts(t *testing.T) {
        data := ints
        Ints(data[0:])
        if !IntsAreSorted(data[0:]) {
                t.Errorf("sorted %v", ints)
                t.Errorf("   got %v", data)
        }
}

func TestFloat64s(t *testing.T) {
        data := float64s
        Float64s(data[0:])
        if !Float64sAreSorted(data[0:]) {
                t.Errorf("sorted %v", float64s)
                t.Errorf("   got %v", data)
        }
}

func TestStrings(t *testing.T) {
        data := strings
        Strings(data[0:])
        if !StringsAreSorted(data[0:]) {
                t.Errorf("sorted %v", strings)
                t.Errorf("   got %v", data)
        }
}

func TestSortLarge_Random(t *testing.T) {
        n := 1000000
        if testing.Short() {
                n /= 100
        }
        data := make([]int, n)
        for i := 0; i < len(data); i++ {
                data[i] = rand.Intn(100)
        }
        if IntsAreSorted(data) {
                t.Fatalf("terrible rand.rand")
        }
        Ints(data)
        if !IntsAreSorted(data) {
                t.Errorf("sort didn't sort - 1M ints")
        }
}

func TestReverseSortIntSlice(t *testing.T) {
        data := ints
        data1 := ints
        a := IntSlice(data[0:])
        Sort(a)
        r := IntSlice(data1[0:])
        Sort(Reverse(r))
        for i := 0; i < len(data); i++ {
                if a[i] != r[len(data)-1-i] {
                        t.Errorf("reverse sort didn't sort")
                }
                if i > len(data)/2 {
                        break
                }
        }
}

type nonDeterministicTestingData struct {
        r *rand.Rand
}

func (t *nonDeterministicTestingData) Len() int {
        return 500
}
func (t *nonDeterministicTestingData) Less(i, j int) bool {
        if i < 0 || j < 0 || i >= t.Len() || j >= t.Len() {
                panic("nondeterministic comparison out of bounds")
        }
        return t.r.Float32() < 0.5
}
func (t *nonDeterministicTestingData) Swap(i, j int) {
        if i < 0 || j < 0 || i >= t.Len() || j >= t.Len() {
                panic("nondeterministic comparison out of bounds")
        }
}

func TestNonDeterministicComparison(t *testing.T) {
        // Ensure that sort.Sort does not panic when Less returns inconsistent results.
        // See https://golang.org/issue/14377.
        defer func() {
                if r := recover(); r != nil {
                        t.Error(r)
                }
        }()

        td := &nonDeterministicTestingData{
                r: rand.New(rand.NewSource(0)),
        }

        for i := 0; i < 10; i++ {
                Sort(td)
        }
}

func BenchmarkSortString1K(b *testing.B) {
        b.StopTimer()
        for i := 0; i < b.N; i++ {
                data := make([]string, 1<<10)
                for i := 0; i < len(data); i++ {
                        data[i] = strconv.Itoa(i ^ 0x2cc)
                }
                b.StartTimer()
                Strings(data)
                b.StopTimer()
        }
}

func BenchmarkStableString1K(b *testing.B) {
        b.StopTimer()
        for i := 0; i < b.N; i++ {
                data := make([]string, 1<<10)
                for i := 0; i < len(data); i++ {
                        data[i] = strconv.Itoa(i ^ 0x2cc)
                }
                b.StartTimer()
                Stable(StringSlice(data))
                b.StopTimer()
        }
}

func BenchmarkSortInt1K(b *testing.B) {
        b.StopTimer()
        for i := 0; i < b.N; i++ {
                data := make([]int, 1<<10)
                for i := 0; i < len(data); i++ {
                        data[i] = i ^ 0x2cc
                }
                b.StartTimer()
                Ints(data)
                b.StopTimer()
        }
}

func BenchmarkStableInt1K(b *testing.B) {
        b.StopTimer()
        for i := 0; i < b.N; i++ {
                data := make([]int, 1<<10)
                for i := 0; i < len(data); i++ {
                        data[i] = i ^ 0x2cc
                }
                b.StartTimer()
                Stable(IntSlice(data))
                b.StopTimer()
        }
}

func BenchmarkSortInt64K(b *testing.B) {
        b.StopTimer()
        for i := 0; i < b.N; i++ {
                data := make([]int, 1<<16)
                for i := 0; i < len(data); i++ {
                        data[i] = i ^ 0xcccc
                }
                b.StartTimer()
                Ints(data)
                b.StopTimer()
        }
}

func BenchmarkStableInt64K(b *testing.B) {
        b.StopTimer()
        for i := 0; i < b.N; i++ {
                data := make([]int, 1<<16)
                for i := 0; i < len(data); i++ {
                        data[i] = i ^ 0xcccc
                }
                b.StartTimer()
                Stable(IntSlice(data))
                b.StopTimer()
        }
}

const (
        _Sawtooth = iota
        _Rand
        _Stagger
        _Plateau
        _Shuffle
        _NDist
)

const (
        _Copy = iota
        _Reverse
        _ReverseFirstHalf
        _ReverseSecondHalf
        _Sorted
        _Dither
        _NMode
)

type testingData struct {
        desc        string
        t           *testing.T
        data        []int
        maxswap     int // number of swaps allowed
        ncmp, nswap int
}

func (d *testingData) Len() int { return len(d.data) }
func (d *testingData) Less(i, j int) bool {
        d.ncmp++
        return d.data[i] < d.data[j]
}
func (d *testingData) Swap(i, j int) {
        if d.nswap >= d.maxswap {
                d.t.Errorf("%s: used %d swaps sorting slice of %d", d.desc, d.nswap, len(d.data))
                d.t.FailNow()
        }
        d.nswap++
        d.data[i], d.data[j] = d.data[j], d.data[i]
}

func min(a, b int) int {
        if a < b {
                return a
        }
        return b
}

func lg(n int) int {
        i := 0
        for 1<<uint(i) < n {
                i++
        }
        return i
}

func testBentleyMcIlroy(t *testing.T, sort func(Interface), maxswap func(int) int) {
        sizes := []int{100, 1023, 1024, 1025}
        if testing.Short() {
                sizes = []int{100, 127, 128, 129}
        }
        dists := []string{"sawtooth", "rand", "stagger", "plateau", "shuffle"}
        modes := []string{"copy", "reverse", "reverse1", "reverse2", "sort", "dither"}
        var tmp1, tmp2 [1025]int
        for _, n := range sizes {
                for m := 1; m < 2*n; m *= 2 {
                        for dist := 0; dist < _NDist; dist++ {
                                j := 0
                                k := 1
                                data := tmp1[0:n]
                                for i := 0; i < n; i++ {
                                        switch dist {
                                        case _Sawtooth:
                                                data[i] = i % m
                                        case _Rand:
                                                data[i] = rand.Intn(m)
                                        case _Stagger:
                                                data[i] = (i*m + i) % n
                                        case _Plateau:
                                                data[i] = min(i, m)
                                        case _Shuffle:
                                                if rand.Intn(m) != 0 {
                                                        j += 2
                                                        data[i] = j
                                                } else {
                                                        k += 2
                                                        data[i] = k
                                                }
                                        }
                                }

                                mdata := tmp2[0:n]
                                for mode := 0; mode < _NMode; mode++ {
                                        switch mode {
                                        case _Copy:
                                                for i := 0; i < n; i++ {
                                                        mdata[i] = data[i]
                                                }
                                        case _Reverse:
                                                for i := 0; i < n; i++ {
                                                        mdata[i] = data[n-i-1]
                                                }
                                        case _ReverseFirstHalf:
                                                for i := 0; i < n/2; i++ {
                                                        mdata[i] = data[n/2-i-1]
                                                }
                                                for i := n / 2; i < n; i++ {
                                                        mdata[i] = data[i]
                                                }
                                        case _ReverseSecondHalf:
                                                for i := 0; i < n/2; i++ {
                                                        mdata[i] = data[i]
                                                }
                                                for i := n / 2; i < n; i++ {
                                                        mdata[i] = data[n-(i-n/2)-1]
                                                }
                                        case _Sorted:
                                                for i := 0; i < n; i++ {
                                                        mdata[i] = data[i]
                                                }
                                                // Ints is known to be correct
                                                // because mode Sort runs after mode _Copy.
                                                Ints(mdata)
                                        case _Dither:
                                                for i := 0; i < n; i++ {
                                                        mdata[i] = data[i] + i%5
                                                }
                                        }

                                        desc := fmt.Sprintf("n=%d m=%d dist=%s mode=%s", n, m, dists[dist], modes[mode])
                                        d := &testingData{desc: desc, t: t, data: mdata[0:n], maxswap: maxswap(n)}
                                        sort(d)
                                        // Uncomment if you are trying to improve the number of compares/swaps.
                                        //t.Logf("%s: ncmp=%d, nswp=%d", desc, d.ncmp, d.nswap)

                                        // If we were testing C qsort, we'd have to make a copy
                                        // of the slice and sort it ourselves and then compare
                                        // x against it, to ensure that qsort was only permuting
                                        // the data, not (for example) overwriting it with zeros.
                                        //
                                        // In go, we don't have to be so paranoid: since the only
                                        // mutating method Sort can call is TestingData.swap,
                                        // it suffices here just to check that the final slice is sorted.
                                        if !IntsAreSorted(mdata) {
                                                t.Errorf("%s: ints not sorted", desc)
                                                t.Errorf("\t%v", mdata)
                                                t.FailNow()
                                        }
                                }
                        }
                }
        }
}

func TestSortBM(t *testing.T) {
        testBentleyMcIlroy(t, Sort, func(n int) int { return n * lg(n) * 12 / 10 })
}

func TestHeapsortBM(t *testing.T) {
        testBentleyMcIlroy(t, Heapsort, func(n int) int { return n * lg(n) * 12 / 10 })
}

func TestStableBM(t *testing.T) {
        testBentleyMcIlroy(t, Stable, func(n int) int { return n * lg(n) * lg(n) / 3 })
}

// This is based on the "antiquicksort" implementation by M. Douglas McIlroy.
// See http://www.cs.dartmouth.edu/~doug/mdmspe.pdf for more info.
type adversaryTestingData struct {
        data      []int
        keys      map[int]int
        candidate int
}

func (d *adversaryTestingData) Len() int { return len(d.data) }

func (d *adversaryTestingData) Less(i, j int) bool {
        if _, present := d.keys[i]; !present {
                if _, present := d.keys[j]; !present {
                        if i == d.candidate {
                                d.keys[i] = len(d.keys)
                        } else {
                                d.keys[j] = len(d.keys)
                        }
                }
        }

        if _, present := d.keys[i]; !present {
                d.candidate = i
                return false
        }
        if _, present := d.keys[j]; !present {
                d.candidate = j
                return true
        }

        return d.keys[i] >= d.keys[j]
}

func (d *adversaryTestingData) Swap(i, j int) {
        d.data[i], d.data[j] = d.data[j], d.data[i]
}

func TestAdversary(t *testing.T) {
        const size = 100
        data := make([]int, size)
        for i := 0; i < size; i++ {
                data[i] = i
        }

        d := &adversaryTestingData{data, make(map[int]int), 0}
        Sort(d) // This should degenerate to heapsort.
}

func TestStableInts(t *testing.T) {
        data := ints
        Stable(IntSlice(data[0:]))
        if !IntsAreSorted(data[0:]) {
                t.Errorf("nsorted %v\n   got %v", ints, data)
        }
}

type intPairs []struct {
        a, b int
}

// IntPairs compare on a only.
func (d intPairs) Len() int           { return len(d) }
func (d intPairs) Less(i, j int) bool { return d[i].a < d[j].a }
func (d intPairs) Swap(i, j int)      { d[i], d[j] = d[j], d[i] }

// Record initial order in B.
func (d intPairs) initB() {
        for i := range d {
                d[i].b = i
        }
}

// InOrder checks if a-equal elements were not reordered.
func (d intPairs) inOrder() bool {
        lastA, lastB := -1, 0
        for i := 0; i < len(d); i++ {
                if lastA != d[i].a {
                        lastA = d[i].a
                        lastB = d[i].b
                        continue
                }
                if d[i].b <= lastB {
                        return false
                }
                lastB = d[i].b
        }
        return true
}

func TestStability(t *testing.T) {
        n, m := 100000, 1000
        if testing.Short() {
                n, m = 1000, 100
        }
        data := make(intPairs, n)

        // random distribution
        for i := 0; i < len(data); i++ {
                data[i].a = rand.Intn(m)
        }
        if IsSorted(data) {
                t.Fatalf("terrible rand.rand")
        }
        data.initB()
        Stable(data)
        if !IsSorted(data) {
                t.Errorf("Stable didn't sort %d ints", n)
        }
        if !data.inOrder() {
                t.Errorf("Stable wasn't stable on %d ints", n)
        }

        // already sorted
        data.initB()
        Stable(data)
        if !IsSorted(data) {
                t.Errorf("Stable shuffled sorted %d ints (order)", n)
        }
        if !data.inOrder() {
                t.Errorf("Stable shuffled sorted %d ints (stability)", n)
        }

        // sorted reversed
        for i := 0; i < len(data); i++ {
                data[i].a = len(data) - i
        }
        data.initB()
        Stable(data)
        if !IsSorted(data) {
                t.Errorf("Stable didn't sort %d ints", n)
        }
        if !data.inOrder() {
                t.Errorf("Stable wasn't stable on %d ints", n)
        }
}

var countOpsSizes = []int{1e2, 3e2, 1e3, 3e3, 1e4, 3e4, 1e5, 3e5, 1e6}

func countOps(t *testing.T, algo func(Interface), name string) {
        sizes := countOpsSizes
        if testing.Short() {
                sizes = sizes[:5]
        }
        if !testing.Verbose() {
                t.Skip("Counting skipped as non-verbose mode.")
        }
        for _, n := range sizes {
                td := testingData{
                        desc:    name,
                        t:       t,
                        data:    make([]int, n),
                        maxswap: 1<<31 - 1,
                }
                for i := 0; i < n; i++ {
                        td.data[i] = rand.Intn(n / 5)
                }
                algo(&td)
                t.Logf("%s %8d elements: %11d Swap, %10d Less", name, n, td.nswap, td.ncmp)
        }
}

func TestCountStableOps(t *testing.T) { countOps(t, Stable, "Stable") }
func TestCountSortOps(t *testing.T)   { countOps(t, Sort, "Sort  ") }

func bench(b *testing.B, size int, algo func(Interface), name string) {
        b.StopTimer()
        data := make(intPairs, size)
        x := ^uint32(0)
        for i := 0; i < b.N; i++ {
                for n := size - 3; n <= size+3; n++ {
                        for i := 0; i < len(data); i++ {
                                x += x
                                x ^= 1
                                if int32(x) < 0 {
                                        x ^= 0x88888eef
                                }
                                data[i].a = int(x % uint32(n/5))
                        }
                        data.initB()
                        b.StartTimer()
                        algo(data)
                        b.StopTimer()
                        if !IsSorted(data) {
                                b.Errorf("%s did not sort %d ints", name, n)
                        }
                        if name == "Stable" && !data.inOrder() {
                                b.Errorf("%s unstable on %d ints", name, n)
                        }
                }
        }
}

func BenchmarkSort1e2(b *testing.B)   { bench(b, 1e2, Sort, "Sort") }
func BenchmarkStable1e2(b *testing.B) { bench(b, 1e2, Stable, "Stable") }
func BenchmarkSort1e4(b *testing.B)   { bench(b, 1e4, Sort, "Sort") }
func BenchmarkStable1e4(b *testing.B) { bench(b, 1e4, Stable, "Stable") }
func BenchmarkSort1e6(b *testing.B)   { bench(b, 1e6, Sort, "Sort") }
func BenchmarkStable1e6(b *testing.B) { bench(b, 1e6, Stable, "Stable") }