summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorRoss Barnowski <rossbar@berkeley.edu>2020-03-06 00:19:32 -0800
committerRoss Barnowski <rossbar@berkeley.edu>2020-03-09 10:39:19 -0700
commitffe1f46121cd11b2b876d20ba1758a09cb4e5be7 (patch)
treecc615cdf9471ee6323184d523082b6a73b2fefae
parentc898ff3f0338f2b24b8f4e7d1b4fffc36d3e51c9 (diff)
downloadnumpy-ffe1f46121cd11b2b876d20ba1758a09cb4e5be7.tar.gz
Added benchmark for _var with complex input.
-rw-r--r--benchmarks/benchmarks/bench_core.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/benchmarks/benchmarks/bench_core.py b/benchmarks/benchmarks/bench_core.py
index 94d3ad503..060d0f7db 100644
--- a/benchmarks/benchmarks/bench_core.py
+++ b/benchmarks/benchmarks/bench_core.py
@@ -180,3 +180,14 @@ class UnpackBits(Benchmark):
class Indices(Benchmark):
def time_indices(self):
np.indices((1000, 500))
+
+class VarComplex(Benchmark):
+ params = [10**n for n in range(1, 9)]
+ def setup(self, n):
+ self.arr = np.random.randn(n) + 1j * np.random.randn(n)
+
+ def teardown(self, n):
+ del self.arr
+
+ def time_var(self, n):
+ self.arr.var()