map64_test.go

package intmap

import (
	"testing"
)

func TestMap64(t *testing.T) {
	type pairs [][2]int64
	cases := []struct {
		name string
		vals pairs
	}{
		{
			name: "empty",
		},
		{
			name: "one",
			vals: pairs{{1, 2}},
		},
		{
			name: "one_zero",
			vals: pairs{{0, 2}},
		},
		{
			name: "two",
			vals: pairs{{1, 2}, {3, 4}},
		},
		{
			name: "two_zero",
			vals: pairs{{1, 2}, {0, 4}},
		},
		{
			name: "ten",
			vals: pairs{{1, 1}, {2, 2}, {3, 3}, {4, 4}, {5, 5}, {6, 6}, {7, 7}, {8, 8}, {9, 9}, {10, 10}},
		},
		{
			name: "ten_zero",
			vals: pairs{{1, 1}, {2, 2}, {3, 3}, {4, 4}, {5, 5}, {6, 6}, {7, 7}, {8, 8}, {9, 9}, {10, 10}, {0, 11}},
		},
	}

	runTest := func(t *testing.T, m *Map[int64, int64], vals pairs) {
		for i, pair := range vals {
			m.Put(pair[0], pair[1])
			if sz := m.Len(); sz != i+1 {
				t.Fatalf("unexpected size after %d put()s: %d", sz, i+1)
			}
		}
		for i, pair := range vals {
			val, ok := m.Get(pair[0])
			if !ok {
				t.Fatalf("key number %d not found: %d", i, pair[0])
			}
			if val != pair[1] {
				t.Fatalf("incorrect value %d for key %d, expected %d", pair[1], pair[0], val)
			}
		}
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Run("zero_cap", func(t *testing.T) {
				m := New[int64, int64](0)
				runTest(t, m, tc.vals)
			})
			t.Run("full_cap", func(t *testing.T) {
				m := New[int64, int64](len(tc.vals))
				runTest(t, m, tc.vals)
			})
		})
	}
}

func TestNilMap(t *testing.T) {
	var m *Map[int, int]

	if sz := m.Len(); sz != 0 {
		t.Fatalf("nil map must have length zero: %d", sz)
	}

	if m.Has(0) || m.Has(1) {
		t.Fatalf("nil map must not have 0 or 1 as keys")
	}

	zero, ok := m.Get(0)
	if ok {
		t.Fatalf("nil map must not have 0 as key")
	}
	if zero != 0 {
		t.Fatalf("nil map must return zero value for missing keys")
	}

	count := 0
	m.ForEach(func(i int, i2 int) bool {
		count++
		t.Fatalf("must not be reached, nil map has no elements")
		return true
	})
	if count != 0 {
		t.Fatalf("iterating over nil map must not yield")
	}

	if m != nil { // sanity check
		t.Fatalf("bad test - m must be nil")
	}
}

func TestMap64Delete(t *testing.T) {
	m := New[int, int](10)
	for i := 0; i < 100; i++ {
		m.Put(i, -i)
	}
	if sz := m.Len(); sz != 100 {
		t.Fatalf("expected %d elements in map: %d", 100, sz)
	}

	for i := 0; i < 100; i++ {
		if found := m.Del(i); !found {
			t.Fatalf("deleted key should have been there: %d", i)
		}
		if sz := m.Len(); sz != 100-i-1 {
			t.Fatalf("expected %d elements in map: %d", 100-i-1, sz)
		}
		if found := m.Del(i); found {
			t.Fatalf("deleted key should not be there: %d", i)
		}
	}

	if sz := m.Len(); sz != 0 {
		t.Fatalf("map not empty, %d elements remain", sz)
	}
}

func TestMap64Has(t *testing.T) {
	m := New[int, int](10)
	for i := 0; i < 100; i++ {
		m.Put(i, -i)
	}
	if sz := m.Len(); sz != 100 {
		t.Fatalf("expected %d elements in map: %d", 100, sz)
	}

	for i := 0; i < 100; i++ {
		if found := m.Has(i); !found {
			t.Fatalf("key should have been there: %d", i)
		}
		if found := m.Has(i + 100); found {
			t.Fatalf("key should not be there: %d", i+100)
		}
	}
}

func TestMap64PutIfNotExists(t *testing.T) {
	m := New[int, int](10)
	for i := 0; i < 100; i++ {
		m.Put(i, -i)
	}
	if sz := m.Len(); sz != 100 {
		t.Fatalf("expected %d elements in map: %d", 100, sz)
	}

	for i := 0; i < 100; i++ {
		val, ok := m.PutIfNotExists(i, i+100)
		if ok {
			t.Fatalf("key should have been there: %d", i)
		}
		if val != -i {
			t.Fatalf("key should have been there: %d", i)
		}
	}
}

func TestMap64ForEachStop(t *testing.T) {
	m := New[int, int](10)
	for i := 0; i < 100; i++ {
		m.Put(i, -i)
	}

	count := 0
	m.ForEach(func(k, v int) bool {
		count++
		return count < 50
	})

	if have, want := count, 50; have != want {
		t.Fatalf("unexpected number of elements processed: %d, want %d", have, want)
	}
}

func TestMap64Iterators(t *testing.T) {
	m := New[int, int](10)
	for i := 0; i < 100; i++ {
		m.Put(i, 99-i) // 0:99, 1:98, 2:97, ...
	}
	const sumTo99 = 99 * (99 + 1) / 2

	sum := 0
	for k, v := range m.All() {
		sum += k + v
	}

	if sum != sumTo99*2 {
		t.Fatalf("unexpected sum when iterating over keys and values: %d, want %d", sum, sumTo99*2)
	}

	sum = 0
	for k := range m.Keys() {
		if k == 9 {
			continue
		}
		sum += k
	}
	expected := sumTo99 - 9
	if sum != expected {
		t.Fatalf("unexpected sum when iterating over keys: %d, want %d", sum, expected)
	}

	sum = 0
	for v := range m.Values() {
		sum += v
	}
	if sum != sumTo99 {
		t.Fatalf("unexpected sum when iterating over values: %d, want %d", sum, sumTo99*2)
	}
}