summaryrefslogtreecommitdiff
path: root/benchmarks
diff options
context:
space:
mode:
authorRohit Goswami <rog32@hi.is>2022-12-03 16:26:50 +0000
committerRohit Goswami <rog32@hi.is>2022-12-03 16:26:50 +0000
commit7c24bd4c36c2793238305f3043c7a71cd89c455f (patch)
tree4f3ff9327638e335e3dde9643c79bdf7db35319f /benchmarks
parent4c4e803c3866fd31440e8db395142eeb2c2ffbd4 (diff)
downloadnumpy-7c24bd4c36c2793238305f3043c7a71cd89c455f.tar.gz
BENCH: Add a test for masked array creations
Diffstat (limited to 'benchmarks')
-rw-r--r--benchmarks/benchmarks/bench_ma.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/benchmarks/benchmarks/bench_ma.py b/benchmarks/benchmarks/bench_ma.py
index b214c0b86..0247065a7 100644
--- a/benchmarks/benchmarks/bench_ma.py
+++ b/benchmarks/benchmarks/bench_ma.py
@@ -17,6 +17,14 @@ class MA(Benchmark):
def time_masked_array_l100_t100(self):
np.ma.masked_array(self.l100, self.t100)
+class MACreation(Benchmark):
+ param_names = ['data', 'mask']
+ params = [[10, 100, 1000],
+ [True, False, None]]
+
+ def time_ma_creations(self, data, mask):
+ np.ma.array(data=np.zeros(int(data)), mask=mask)
+
class Indexing(Benchmark):
param_names = ['masked', 'ndim', 'size']