From 80227f5ccd3c1e21fcef04fa2ae9df4840b2bfa6 Mon Sep 17 00:00:00 2001
From: Alexander Sikorski <sikorski@zib.de>
Date: Wed, 10 Jun 2020 15:56:26 +0200
Subject: [PATCH] add benchmarks

---
 tests/test_pcca.py | 21 +++++++++++++++++++++
 tox.ini            |  3 ++-
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/tests/test_pcca.py b/tests/test_pcca.py
index 3b53dd4..d844b1f 100644
--- a/tests/test_pcca.py
+++ b/tests/test_pcca.py
@@ -49,3 +49,24 @@ def test_krylovschur(n=30, m=5, N=100):
             K = pcca.krylovschur(A, m)
             R = np.linalg.matrix_rank(np.concatenate([S, K], axis=1), tol=1e-6)
             assert R == m
+
+
+def test_bench_scipyschur(benchmark, n=500, m=6):
+    T = utils.randompropagator(n)
+    massmatrix = np.diag(np.ones(n))
+    solver = pcca.ScipySchur()
+    benchmark(solver.solve, T, m, massmatrix)
+
+
+def test_bench_scipyqz(benchmark, n=500, m=6):
+    T = utils.randompropagator(n)
+    solver = pcca.ScipySchur()
+    benchmark(solver.solve, T, m)
+
+
+def test_bench_krylovschur(benchmark, n=500, m=6):
+    if not pcca.HAS_SLEPC:
+        return
+    T = utils.randompropagator(n)
+    solver = pcca.KrylovSchur(onseperation="continue")
+    benchmark(solver.solve, T, m)
diff --git a/tox.ini b/tox.ini
index b12f2c9..9fe545b 100644
--- a/tox.ini
+++ b/tox.ini
@@ -10,5 +10,6 @@ envlist = py37, py27
 deps =
     pytest
     pytest-cov
+    pytest-benchmark
 commands =
-    pytest --cov=cmdtools --cov-report term-missing
+    pytest --cov=cmdtools --cov-report term-missing --benchmark-skip