From 0b2cbd28e2f47ac7d0831a139b9543cb19ba936f Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Tue, 30 Aug 2016 04:05:06 +0200 Subject: Solve Nth prime --- go/nth-prime/nth_prime_test.go | 39 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) create mode 100644 go/nth-prime/nth_prime_test.go (limited to 'go/nth-prime/nth_prime_test.go') diff --git a/go/nth-prime/nth_prime_test.go b/go/nth-prime/nth_prime_test.go new file mode 100644 index 0000000..03f23e8 --- /dev/null +++ b/go/nth-prime/nth_prime_test.go @@ -0,0 +1,39 @@ +package prime + +import "testing" + +var tests = []struct { + n int + p int + ok bool +}{ + {1, 2, true}, + {2, 3, true}, + {3, 5, true}, + {4, 7, true}, + {5, 11, true}, + {6, 13, true}, + {10001, 104743, true}, + {0, 0, false}, +} + +func TestNth(t *testing.T) { + for _, test := range tests { + switch p, ok := Nth(test.n); { + case !ok: + if test.ok { + t.Fatalf("Nth(%d) returned !ok. Expecting ok.", test.n) + } + case !test.ok: + t.Fatalf("Nth(%d) = %d, ok = %t. Expecting !ok.", test.n, p, ok) + case p != test.p: + t.Fatalf("Nth(%d) = %d, want %d.", test.n, p, test.p) + } + } +} + +func BenchmarkNth(b *testing.B) { + for i := 0; i < b.N; i++ { + Nth(10001) + } +} -- cgit v1.2.3