diff --git a/src/saving_test.go b/src/saving_test.go index 86a9378..0434b0d 100644 --- a/src/saving_test.go +++ b/src/saving_test.go @@ -89,3 +89,35 @@ func BenchmarkJSONSaving(b *testing.B) { SaveToJSONWriter(data, keyMap, tmpFile, true) } + +func BenchmarkCsvGzSaving(b *testing.B) { + data := GenerateRandStatsForTime(b.N) + keyMap := GetKeymap() + + tmpFile, err := ioutil.TempFile(os.TempDir(), "benchmark") + if err != nil { + b.Fatal(err) + } + + defer os.Remove(tmpFile.Name()) + + b.ResetTimer() + + SaveToCsvGzWriter(data, keyMap, tmpFile, true) +} + +func BenchmarkJSONGzSaving(b *testing.B) { + data := GenerateRandStatsForTime(b.N) + keyMap := GetKeymap() + + tmpFile, err := ioutil.TempFile(os.TempDir(), "benchmark") + if err != nil { + b.Fatal(err) + } + + defer os.Remove(tmpFile.Name()) + + b.ResetTimer() + + SaveToJSONGzWriter(data, keyMap, tmpFile, true) +}