21 lines
539 B
Go
21 lines
539 B
Go
package top_k_frequent_elements_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"git.maximhutz.com/practice/pkg/questions/top_k_frequent_elements"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test1(t *testing.T) {
|
|
assert.Equal(t, []int{1, 2}, top_k_frequent_elements.TopKFrequent([]int{1, 1, 1, 2, 2, 3}, 2))
|
|
}
|
|
|
|
func Test2(t *testing.T) {
|
|
assert.Equal(t, []int{1}, top_k_frequent_elements.TopKFrequent([]int{1}, 1))
|
|
}
|
|
|
|
func Test3(t *testing.T) {
|
|
assert.Equal(t, []int{1, 4}, top_k_frequent_elements.TopKFrequent([]int{1, 4, 1, 4, 1, 4, 3, 1, 3, 4}, 2))
|
|
}
|