Merge pull request #175 from rlizzo/asv-update

Benchmark Updates
tensorwerk · Dec 5, 2019 · d267c0a · d267c0a
2 parents 5770f5a + ade181b
commit d267c0a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/asv_bench/benchmarks/backend_comparisons.py b/asv_bench/benchmarks/backend_comparisons.py
@@ -15,7 +15,7 @@ class _WriterSuite:
     params = ['hdf5_00', 'hdf5_01', 'numpy_10']
     param_names = ['backend']
     processes = 2
-    repeat = (2, 2, 20.0)
+    repeat = (3, 3, 40.0)
     # repeat == tuple (min_repeat, max_repeat, max_time)
     number = 1
     warmup_time = 0
@@ -50,8 +50,11 @@ def setup(self, backend):
             aset = self.co.arraysets.init_arrayset(
                 'aset', prototype=arr, backend_opts=self.backend_code[backend])
         except TypeError:
-            aset = self.co.arraysets.init_arrayset(
-                'aset', prototype=arr, backend=self.backend_code[backend])
+            try:
+                aset = self.co.arraysets.init_arrayset(
+                    'aset', prototype=arr, backend=self.backend_code[backend])
+            except ValueError:
+                raise NotImplementedError
         except ValueError:
             raise NotImplementedError
 

diff --git a/asv_bench/benchmarks/backends/hdf5_01.py b/asv_bench/benchmarks/backends/hdf5_01.py
@@ -10,7 +10,7 @@
 class _WriterSuite_HDF5_01:
 
     processes = 2
-    repeat = (2, 2, 10.0)
+    repeat = (2, 2, 20.0)
     # repeat == tuple (min_repeat, max_repeat, max_time)
     number = 1
     warmup_time = 0
@@ -38,7 +38,10 @@ def setup(self):
         try:
             aset = self.co.arraysets.init_arrayset('aset', prototype=arr, backend_opts='01')
         except TypeError:
-            aset = self.co.arraysets.init_arrayset('aset', prototype=arr, backend='01')
+            try:
+                aset = self.co.arraysets.init_arrayset('aset', prototype=arr, backend='01')
+            except ValueError:
+                raise NotImplementedError
         except ValueError:
             # marks as skipped benchmark for commits which do not have this backend.
             raise NotImplementedError