add singer var benchmark

ringsaturn · Aug 11, 2024 · 9000b95 · 9000b95
1 parent 109c87e
commit 9000b95
Showing 1 changed file with 45 additions and 26 deletions.
diff --git a/mep/mep_test.go b/mep/mep_test.go
@@ -171,36 +171,41 @@ func BenchmarkAlgoCalc(b *testing.B) {
 	}
 }
 
-var largeData [][]*goaqi.Var
+var (
+	largeData [][]*goaqi.Var
+	lessData  [][]*goaqi.Var
+)
 
 func init() {
 	for i := 0; i < 384; i++ {
-		largeData = append(largeData, []*goaqi.Var{
-			{
-				P:     goaqi.PM2_5_1H,
-				Value: 16,
-			},
-			{
-				P:     goaqi.PM10_1H,
-				Value: 88,
-			},
-			{
-				P:     goaqi.CO_1H,
-				Value: 0.2,
-			},
-			{
-				P:     goaqi.SO2_1H,
-				Value: 3,
-			},
-			{
-				P:     goaqi.NO2_1H,
-				Value: 11,
-			},
-			{
-				P:     goaqi.O3_1H,
-				Value: 75,
+		largeData = append(
+			largeData,
+			[]*goaqi.Var{
+				{
+					P:     goaqi.PM2_5_1H,
+					Value: 16,
+				},
+				{
+					P:     goaqi.PM10_1H,
+					Value: 88,
+				},
+				{
+					P:     goaqi.CO_1H,
+					Value: 0.2,
+				},
+				{
+					P:     goaqi.SO2_1H,
+					Value: 3,
+				},
+				{
+					P:     goaqi.NO2_1H,
+					Value: 11,
+				},
+				{
+					P:     goaqi.O3_1H,
+					Value: 75,
+				},
 			},
-		},
 		)
 	}
 }
@@ -215,3 +220,17 @@ func BenchmarkAlgoCalcOnLargeData(b *testing.B) {
 		}()
 	}
 }
+
+func BenchmarkAlgoCalcOnSingleDataset(b *testing.B) {
+	algo := &mep.Algo{}
+	for i := 0; i < b.N; i++ {
+		func() {
+			_, _, _ = algo.Calc(
+				&goaqi.Var{
+					P:     goaqi.PM10_1H,
+					Value: 88,
+				},
+			)
+		}()
+	}
+}