day 94: DRY the creation of the test tree

vaskoz · Nov 27, 2018 · a2a27f5 · a2a27f5
1 parent 7de8f90
commit a2a27f5
Showing 1 changed file with 18 additions and 34 deletions.
diff --git a/day94/problem_test.go b/day94/problem_test.go
@@ -2,49 +2,33 @@ package day94
 
 import "testing"
 
-func TestMaxPathSum(t *testing.T) {
-	t.Parallel()
-	tree := &BinaryTree{
-		-15,
-		&BinaryTree{5,
-			&BinaryTree{-8,
-				&BinaryTree{2, nil, nil},
-				&BinaryTree{6, nil, nil}},
-			&BinaryTree{1, nil, nil}},
-		&BinaryTree{6,
-			&BinaryTree{3, nil, nil},
-			&BinaryTree{9, nil,
-				&BinaryTree{0,
-					&BinaryTree{4, nil, nil},
-					&BinaryTree{-1, &BinaryTree{10, nil, nil}, nil},
-				},
+var tree = &BinaryTree{
+	-15,
+	&BinaryTree{5,
+		&BinaryTree{-8,
+			&BinaryTree{2, nil, nil},
+			&BinaryTree{6, nil, nil}},
+		&BinaryTree{1, nil, nil}},
+	&BinaryTree{6,
+		&BinaryTree{3, nil, nil},
+		&BinaryTree{9, nil,
+			&BinaryTree{0,
+				&BinaryTree{4, nil, nil},
+				&BinaryTree{-1, &BinaryTree{10, nil, nil}, nil},
 			},
 		},
-	}
+	},
+}
+
+func TestMaxPathSum(t *testing.T) {
+	t.Parallel()
 	result := MaxPathSum(tree)
 	if result != 27 {
 		t.Errorf("Expected 27 got %v", result)
 	}
 }
 
 func BenchmarkMaxPathSum(b *testing.B) {
-	tree := &BinaryTree{
-		-15,
-		&BinaryTree{5,
-			&BinaryTree{-8,
-				&BinaryTree{2, nil, nil},
-				&BinaryTree{6, nil, nil}},
-			&BinaryTree{1, nil, nil}},
-		&BinaryTree{6,
-			&BinaryTree{3, nil, nil},
-			&BinaryTree{9, nil,
-				&BinaryTree{0,
-					&BinaryTree{4, nil, nil},
-					&BinaryTree{-1, &BinaryTree{10, nil, nil}, nil},
-				},
-			},
-		},
-	}
 	for i := 0; i < b.N; i++ {
 		MaxPathSum(tree)
 	}