median_in_a_stream_test.go

package heap

import "testing"

/*
TestMedianInAStream tests solution(s) with the following signatures and problem description:

	func newMedianKeeper() medianKeeper
	func (m *medianKeeper) addNumber(num int)

Given a stream of integers, return the median of the set after each input.

For example given {1,2,3,4}, return {1, 1.5, 2, 2.5} after each input respectively.
*/
func TestMedianInAStream(t *testing.T) {
	tests := []struct {
		numbers []int
		median  float64
	}{
		{[]int{1}, 1},
		{[]int{2, 3, 4}, 3},
		{[]int{1, 2, 3, 4, 5}, 3},
		{[]int{1, 2, 3, 4, 5, 6}, 3.5},
		{[]int{6, 5, 4, 3, 2, 1}, 3.5},
		{[]int{1, 4, 5, -2, 4, 6}, 4},
	}

	for i, test := range tests {
		heap := newMedianKeeper()
		for _, number := range test.numbers {
			heap.addNumber(number)
		}
		got := heap.median()
		if got != test.median {
			t.Fatalf("Failed test case #%d. Want %v got %v", i, test.median, got)
		}
	}
}

func TestHeap(t *testing.T) {
	minHeap := new(minHeap)
	minHeap.Push(1)
	got := minHeap.Pop().(int)
	if got != 1 {
		t.Fatalf("expected %d got %d", got, 1)
	}
}