瀏覽代碼

Added tests

Piotr Czajkowski 2 年之前
父節點
當前提交
2e01ce1bfa
共有 1 個文件被更改,包括 25 次插入0 次删除
  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")
+	}
+}