Wordcount - Go Guide #43

wordcount.go

package wordcount

import "strings"

func WordCount(s string) map[string]int {
	var (
		arrOfWords []string
		wordMap    map[string]int
	)

	wordMap = make(map[string]int)
	arrOfWords = strings.Fields(s)

	for _, v := range arrOfWords {
		count := wordMap[v]
		wordMap[v] = count + 1
	}

	return wordMap
}

wordcount_test.go

package wordcount

import (
	"testing"

	"github.com/bmizerany/assert"
)

func TestWordCount(t *testing.T) {
	w := WordCount("foo bar bar foo foo foo")
	var s = map[string]int{
		"foo": 4,
		"bar": 2,
	}
	assert.Equal(t, w, s)
}