Add benchmark function

Author: urschrei

pyproject.toml

@@ -36,7 +36,7 @@ features = ["pyo3/extension-module"]
 
 [project.optional-dependencies]
 test = [
-    "pytest >= 7.4.2",
+    "pytest >= 7.4.2", "pyperf >= 2.7.0"
 ]
 [tool.maturin.dynamic]
 readme = {file = "README.md", content-type = "text/markdown"}

src/lib.rs

@@ -67,7 +67,7 @@ impl CVM {
 }
 
 #[pymodule]
-fn cvmcount(_py: Python, m: Bound<'_, PyModule>) -> PyResult<()> {
+fn count_distinct(_py: Python, m: Bound<'_, PyModule>) -> PyResult<()> {
     m.add_class::<CVM>()?;
     // m.add_function(wrap_pyfunction!(version, m)?)?;
 

tests/bench.py

@@ -0,0 +1,23 @@
+import numpy as np
+import pyperf
+from count_distinct import CVM
+
+
+def add_elems(counter, elems):
+    for elem in elems:
+        counter.add(elem)
+
+
+# Generate 10 million random 7-digit integers
+random_integers = np.random.randint(low=1000000, high=10000000, size=10_000_000)
+
+counter = CVM(0.8, 0.1, 100000)
+
+runner = pyperf.Runner()
+runner.bench_func(
+    "10 million 7-digit random positive integers",
+    add_elems,
+    counter,
+    random_integers,
+)
+count = counter.calculate_final_result()