Add benchmark.

neguse · Nov 27, 2021 · 5457112 · 5457112
1 parent 4afb605
commit 5457112
Show file tree

Hide file tree

Showing 3 changed files with 64 additions and 0 deletions.
diff --git a/goi.png b/goi.png
diff --git a/goi.qoi b/goi.qoi
diff --git a/goi_test.go b/goi_test.go
@@ -81,3 +81,67 @@ func TestEncode(t *testing.T) {
 		}
 	}
 }
+
+func BenchmarkDecodePng(b *testing.B) {
+	data, err := os.ReadFile("goi.png")
+	if err != nil {
+		b.Fatal(err)
+	}
+
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		if _, err := png.Decode(bytes.NewReader(data)); err != nil {
+			b.Fatal(err)
+		}
+	}
+}
+
+func BenchmarkDecodeQoi(b *testing.B) {
+	data, err := os.ReadFile("goi.qoi")
+	if err != nil {
+		b.Fatal(err)
+	}
+
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		if _, err := Decode(bytes.NewReader(data)); err != nil {
+			b.Fatal(err)
+		}
+	}
+}
+
+func BenchmarkEncodePng(b *testing.B) {
+	data, err := os.ReadFile("goi.png")
+	if err != nil {
+		b.Fatal(err)
+	}
+	m, err := png.Decode(bytes.NewReader(data))
+	if err != nil {
+		b.Fatal(err)
+	}
+
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		if png.Encode(bytes.NewBuffer(nil), m); err != nil {
+			b.Fatal(err)
+		}
+	}
+}
+
+func BenchmarkEncodeQoi(b *testing.B) {
+	data, err := os.ReadFile("goi.png")
+	if err != nil {
+		b.Fatal(err)
+	}
+	m, err := png.Decode(bytes.NewReader(data))
+	if err != nil {
+		b.Fatal(err)
+	}
+
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		if Encode(bytes.NewBuffer(nil), m); err != nil {
+			b.Fatal(err)
+		}
+	}
+}