diff options
Diffstat (limited to 'benchmarks')
| -rw-r--r-- | benchmarks/benchmarks/bench_core.py | 18 |
1 files changed, 11 insertions, 7 deletions
diff --git a/benchmarks/benchmarks/bench_core.py b/benchmarks/benchmarks/bench_core.py index e96b490ff..ac28e5b81 100644 --- a/benchmarks/benchmarks/bench_core.py +++ b/benchmarks/benchmarks/bench_core.py @@ -11,7 +11,9 @@ class Core(Benchmark): self.l50 = range(50) self.l = [np.arange(1000), np.arange(1000)] self.l10x10 = np.ones((10, 10)) - self.arr = np.arange(10 ** 6)[::-1] + random = np.random.RandomState(123) + self.arr = np.arange(10 ** 4) + random.shuffle(self.arr) def time_array_1(self): np.array(1) @@ -76,14 +78,16 @@ class Core(Benchmark): def time_tril_l10x10(self): np.tril(self.l10x10) - def time_sort(self): - np.sort(self.arr) - def time_quicksort(self): - np.sort(self.arr, kind='q') + np.sort(self.arr, kind='quick') + + def time_sorts(self, kind): + try: + np.sort(self.arr, kind=kind) + except TypeError: + raise NotImplementedError('Sort kind not implemented or invalid dtype') - def time_radixsort(self): - np.sort(self.arr, kind='r') + time_sorts.params = ['quick', 'merge', 'heap', 'radix'] class Temporaries(Benchmark): |