Refactor benchmarks, add sum and harmonic

Author: skirpichev

bench/README.rst

@@ -5,6 +5,7 @@ It's possible to run them also with gmpy2's and flint's integer types:
 .. code:: sh
 
     ( export T="gmpy2.mpz"; \
+      export PYTHONPATH=. ; \
       python bench/mul.py -q --copy-env --rigorous -o $T.json )
 
 Beware, that the gmp prefers clang over gcc and extensions might

bench/collatz.py

@@ -1,17 +1,8 @@
-# collatz.py
-
-import os
+# bench/collatz.py
 
 import pyperf
 
-if os.getenv("T") == "gmpy2.mpz":
-    from gmpy2 import mpz
-elif os.getenv("T") == "flint.fmpz":
-    from flint import fmpz as mpz
-elif os.getenv("T") == "int":
-    mpz = int
-else:
-    from gmp import mpz
+from bench.utils import mpz
 
 zero = mpz(0)
 one = mpz(1)

bench/harmonic.py

@@ -0,0 +1,12 @@
+# bench/harmonic.py
+
+from fractions import Fraction
+
+import pyperf
+
+from bench.utils import mpz, mysum
+
+runner = pyperf.Runner()
+for n in [100, 1000]:
+    xs = [Fraction(mpz(1), mpz(i)) for i in range(1, n + 1)]
+    runner.bench_func(f"H({n})", mysum, xs)

bench/mul.py

@@ -1,18 +1,10 @@
-# mul.py
+# bench/mul.py
 
-import os
 from operator import mul
 
 import pyperf
 
-if os.getenv("T") == "gmpy2.mpz":
-    from gmpy2 import mpz
-elif os.getenv("T") == "flint.fmpz":
-    from flint import fmpz as mpz
-elif os.getenv("T") == "int":
-    mpz = int
-else:
-    from gmp import mpz
+from bench.utils import mpz
 
 values = ["1<<7", "1<<38", "1<<300", "1<<3000"]
 

bench/sum.py

@@ -0,0 +1,10 @@
+# bench/sum.py
+
+import pyperf
+
+from bench.utils import mpz, mysum
+
+runner = pyperf.Runner()
+for n in [100, 1000]:
+    xs = [mpz(i) for i in range(1, n + 1)]
+    runner.bench_func(f"sum({n})", mysum, xs)

bench/utils.py

@@ -0,0 +1,17 @@
+import os
+
+if os.getenv("T") == "gmpy2.mpz":
+    from gmpy2 import mpz
+elif os.getenv("T") == "flint.fmpz":
+    from flint import fmpz as mpz
+elif os.getenv("T") == "int":
+    mpz = int
+else:
+    pass
+
+
+def mysum(xs):
+    total = xs[0]
+    for t in xs[1:]:
+        total += t
+    return t