1 files changed, 7 insertions, 25 deletions
diff --git a/merkle_test.go b/merkle_test.go
index adc8861..c10d17c 100644
--- a/merkle_test.go
+++ b/merkle_test.go
@@ -41,38 +41,20 @@ func TestMerkle(t *testing.T) {
 	}
 }
 
-func TestBenchmarkBFSMerkle(t *testing.T) {
+var proofs [][][]byte
+
+func TestProofsBFSMerkle(t *testing.T) {
 	m := NewBFSMerkle(25, "/mnt/data/bfs.db")
-	var proof [][]byte
+	root := make([]byte, hashSize)
+	m.Read(root, 0)
 	start := time.Now()
-	for i := 0; i < N; i++ {
-		proof = m.Proof(ids[i])
-		if len(proof) != int(m.height) {
-			fmt.Println("error")
-		}
-	}
+	proofs = m.Proofs(ids)
 	elapsed := time.Since(start)
 	fmt.Println(elapsed, elapsed.Nanoseconds()/N)
 }
 
-func TestBenchmarkPostMerkle(t *testing.T) {
+func TestProofsPostMerkle(t *testing.T) {
 	m := NewPostMerkle(25, "/mnt/data/post.db")
-	var proof [][]byte
-	start := time.Now()
-	for i := 0; i < N; i++ {
-		proof = m.Proof(ids[i])
-		if len(proof) != int(m.height) {
-			fmt.Println("error")
-		}
-	}
-	elapsed := time.Since(start)
-	fmt.Println(elapsed, elapsed.Nanoseconds()/N)
-}
-
-var proofs [][][]byte
-
-func TestProofsBFSMerkle(t *testing.T) {
-	m := NewBFSMerkle(25, "/mnt/data/bfs.db")
 	root := make([]byte, hashSize)
 	m.Read(root, 0)
 	start := time.Now()