Add benchmark code for vcf reader

chrovis · Nov 15, 2023 · bab0ae8 · bab0ae8
1 parent 45d60a8
commit bab0ae8
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/bench/cljam/io/vcf_bench.clj b/bench/cljam/io/vcf_bench.clj
@@ -0,0 +1,19 @@
+(ns cljam.io.vcf-bench
+  (:require [cljam.io.vcf :as vcf]
+            [cljam.test-common :as tcommon]
+            [libra.bench :refer [are defbench]]
+            [libra.criterium :as c]))
+
+(defbench decode-small-bcf-bench
+  (are [f]
+       (c/quick-bench
+        (with-open [r (vcf/reader f)]
+          (run! (constantly nil) (vcf/read-variants r))))
+    tcommon/test-bcf-complex-file))
+
+(defbench decode-large-bcf-bench
+  (are [f]
+       (c/quick-bench
+        (with-open [r (vcf/reader f)]
+          (run! (constantly nil) (vcf/read-variants-randomly r {:chr "chr1" :end 30000000} {}))))
+    tcommon/test-large-bcf-file))