diff --git a/2021/day02/main_test.go b/2021/day02/main_test.go
new file mode 100644
index 0000000..263db75
--- /dev/null
+++ b/2021/day02/main_test.go
@@ -0,0 +1,28 @@
+package main
+
+import "testing"
+
+var testInput = []string{
+	"forward 5",
+	"down 5",
+	"forward 8",
+	"up 3",
+	"down 8",
+	"forward 2",
+}
+
+func TestPartOne(t *testing.T) {
+	expected := 150
+	got := PartOne(testInput)
+	if got != expected {
+		t.Errorf("PartOne() = %d, want %d", got, expected)
+	}
+}
+
+func TestPartTwo(t *testing.T) {
+	expected := 900
+	got := PartTwo(testInput)
+	if got != expected {
+		t.Errorf("PartTwo() = %d, want %d", got, expected)
+	}
+}