add test for tuples

nschloe · Jan 31, 2022 · 437faea · 437faea
1 parent a614048
commit 437faea
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/test_perfplot.py b/tests/test_perfplot.py
@@ -135,3 +135,18 @@ def test_no_setup():
 def test_n_setups():
     setups = [np.random.rand] * len(kernels)
     perfplot.show(setup=setups, kernels=kernels, n_range=r)
+
+
+def test_return_tuples():
+    def setup(n):
+        return np.ones(n), np.full(n + 2, 3.1)
+
+    def times(a, b):
+        return 2 * a, 3 * b
+
+    def times_reversed(a, b):
+        return a * 2, b * 3
+
+    perfplot.show(
+        setup=setup, kernels=[times, times_reversed], n_range=[2**k for k in range(3)]
+    )