[query] Add a new benchmark testing speed of collect (#10972)

* Add a new benchmark testing speed of collect * Rename and increase size
hail-is · Oct 14, 2021 · d427851 · d427851
1 parent 1613273
commit d427851
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/benchmark/python/benchmark_hail/run/matrix_table_benchmarks.py b/benchmark/python/benchmark_hail/run/matrix_table_benchmarks.py
@@ -371,3 +371,10 @@ def matrix_multi_write_nothing():
         mt = hl.utils.range_matrix_table(1, 1, n_partitions=1)
         mts = [mt] * 1000
         hl.experimental.write_matrix_tables(mts, path.join(tmpdir, 'multi-write'), overwrite=True)
+
+
+@benchmark(args=profile_25.handle("mt"))
+def mt_localize_and_collect(mt_path):
+    mt = hl.read_matrix_table(mt_path)
+    ht = mt.localize_entries("ent")
+    collected = ht.head(150).collect()