MAINT: Move ma benchmarks to new file

eric-wieser · eric-wieser · commit d5cc7cdc8df4 · 2017-04-11T22:52:10.000+01:00
diff --git a/benchmarks/benchmarks/bench_core.py b/benchmarks/benchmarks/bench_core.py
@@ -96,21 +96,6 @@ def time_large2(self):
         (self.alarge + self.blarge) - 2
 
 
-class MA(Benchmark):
-    def setup(self):
-        self.l100 = range(100)
-        self.t100 = ([True] * 100)
-
-    def time_masked_array(self):
-        np.ma.masked_array()
-
-    def time_masked_array_l100(self):
-        np.ma.masked_array(self.l100)
-
-    def time_masked_array_l100_t100(self):
-        np.ma.masked_array(self.l100, self.t100)
-
-
 class CorrConv(Benchmark):
     params = [[50, 1000, 1e5],
               [10, 100, 1000, 1e4],
diff --git a/benchmarks/benchmarks/bench_ma.py b/benchmarks/benchmarks/bench_ma.py
@@ -0,0 +1,20 @@
+from __future__ import absolute_import, division, print_function
+
+from .common import Benchmark
+
+import numpy as np
+
+
+class MA(Benchmark):
+    def setup(self):
+        self.l100 = range(100)
+        self.t100 = ([True] * 100)
+
+    def time_masked_array(self):
+        np.ma.masked_array()
+
+    def time_masked_array_l100(self):
+        np.ma.masked_array(self.l100)
+
+    def time_masked_array_l100_t100(self):
+        np.ma.masked_array(self.l100, self.t100)
