Browse Source

Added tests

Piotr Czajkowski 2 years ago
parent
commit
2e01ce1bfa
1 changed files with 25 additions and 0 deletions
  1. 25 0
      07/code_test.go

+ 25 - 0
07/code_test.go

@@ -0,0 +1,25 @@
+package main
+
+import "testing"
+
+func TestPart1(t *testing.T) {
+	crabs, orderedCrabs := readInput("./testinput")
+	if len(crabs) == 0 || len(orderedCrabs) == 0 {
+		t.Error("Could not read input")
+	}
+
+	if calculate(crabs, orderedCrabs, costPart1) != 37 {
+		t.Error("Part 1 failed")
+	}
+}
+
+func TestPart2(t *testing.T) {
+	crabs, orderedCrabs := readInput("./testinput")
+	if len(crabs) == 0 || len(orderedCrabs) == 0 {
+		t.Error("Could not read input")
+	}
+
+	if calculate(crabs, orderedCrabs, costPart2) != 168 {
+		t.Error("Part 2 failed")
+	}
+}