From 1f7721e922441e76654f302f60238c3a864e92a8 Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Fri, 26 Aug 2016 11:19:33 +0200 Subject: Add ETL problem --- go/etl/etl_test.go | 90 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 90 insertions(+) create mode 100644 go/etl/etl_test.go (limited to 'go/etl/etl_test.go') diff --git a/go/etl/etl_test.go b/go/etl/etl_test.go new file mode 100644 index 0000000..d52af10 --- /dev/null +++ b/go/etl/etl_test.go @@ -0,0 +1,90 @@ +package etl + +import "testing" + +type given map[int][]string +type expectation map[string]int + +var transformTests = []struct { + input given + output expectation +}{ + { + given{1: {"A"}}, + expectation{"a": 1}, + }, + { + given{1: {"A", "E", "I", "O", "U"}}, + expectation{"a": 1, "e": 1, "i": 1, "o": 1, "u": 1}, + }, + { + given{ + 1: {"A", "E"}, + 2: {"D", "G"}, + }, + expectation{ + "a": 1, + "e": 1, + "d": 2, + "g": 2, + }, + }, + { + given{ + 1: {"A", "E", "I", "O", "U", "L", "N", "R", "S", "T"}, + 2: {"D", "G"}, + 3: {"B", "C", "M", "P"}, + 4: {"F", "H", "V", "W", "Y"}, + 5: {"K"}, + 8: {"J", "X"}, + 10: {"Q", "Z"}, + }, + expectation{ + "a": 1, "e": 1, "i": 1, "o": 1, "u": 1, "l": 1, "n": 1, "r": 1, "s": 1, "t": 1, + "d": 2, "g": 2, + "b": 3, "c": 3, "m": 3, "p": 3, + "f": 4, "h": 4, "v": 4, "w": 4, "y": 4, + "k": 5, + "j": 8, "x": 8, + "q": 10, "z": 10, + }, + }, +} + +func equal(actual map[string]int, expectation map[string]int) bool { + if len(actual) != len(expectation) { + return false + } + + for k, actualVal := range actual { + expectationVal, present := expectation[k] + + if !present || actualVal != expectationVal { + return false + } + } + + return true +} + +func TestTransform(t *testing.T) { + for _, tt := range transformTests { + actual := Transform(map[int][]string(tt.input)) + if !equal(actual, tt.output) { + t.Fatalf("Transform(%v). Expected [%v], Actual [%v]", tt.input, tt.output, actual) + } + } +} + +func BenchmarkTransform(b *testing.B) { + b.StopTimer() + for _, tt := range transformTests { + b.StartTimer() + + for i := 0; i < b.N; i++ { + Transform(map[int][]string(tt.input)) + } + + b.StopTimer() + } +} -- cgit v1.2.3