Merge pull request #22726 from HaoZeke/benchMA

BENCH: Add a test for masked array creations

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']