File: deque_test.go

package info (click to toggle)
golang-github-cilium-ebpf 0.17.3%2Bds1-1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 4,684 kB
  • sloc: ansic: 1,259; makefile: 127; python: 113; awk: 29; sh: 24
file content (82 lines) | stat: -rw-r--r-- 1,353 bytes parent folder | download | duplicates (2)
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package internal

import "testing"

func TestDeque(t *testing.T) {
	t.Run("pop", func(t *testing.T) {
		var dq Deque[int]
		dq.Push(1)
		dq.Push(2)

		if dq.Pop() != 2 {
			t.Error("Didn't pop 2 first")
		}

		if dq.Pop() != 1 {
			t.Error("Didn't pop 1 second")
		}

		if dq.Pop() != 0 {
			t.Error("Didn't pop zero")
		}
	})

	t.Run("shift", func(t *testing.T) {
		var td Deque[int]
		td.Push(1)
		td.Push(2)

		if td.Shift() != 1 {
			t.Error("Didn't shift 1 first")
		}

		if td.Shift() != 2 {
			t.Error("Didn't shift b second")
		}

		if td.Shift() != 0 {
			t.Error("Didn't shift zero")
		}
	})

	t.Run("push", func(t *testing.T) {
		var td Deque[int]
		td.Push(1)
		td.Push(2)
		td.Shift()

		for i := 1; i <= 12; i++ {
			td.Push(i)
		}

		if td.Shift() != 2 {
			t.Error("Didn't shift 2 first")
		}
		for i := 1; i <= 12; i++ {
			if v := td.Shift(); v != i {
				t.Fatalf("Shifted %d at pos %d", v, i)
			}
		}
	})

	t.Run("grow", func(t *testing.T) {
		var td Deque[int]
		td.Push(1)
		td.Push(2)
		td.Push(3)
		td.Shift()

		td.Grow(7)
		if len(td.elems) < 9 {
			t.Fatal("Expected at least 9 elements, got", len(td.elems))
		}

		if cap(td.elems)&(cap(td.elems)-1) != 0 {
			t.Fatalf("Capacity %d is not a power of two", cap(td.elems))
		}

		if td.Shift() != 2 || td.Shift() != 3 {
			t.Fatal("Elements don't match after grow")
		}
	})
}