Merge 71ee3e4 into 5fba799

tests_perf/common.py

@@ -0,0 +1,31 @@
+from enum import Enum
+
+import numba
+
+import hpat
+
+
+class Implementation(Enum):
+    native = 'native'
+    njit = 'njit'
+    hpat = 'hpat'
+
+
+class ImplRunner:
+    def __init__(self, implementation, func):
+        self.implementation = implementation
+        self.func = func
+
+    @property
+    def runner(self):
+        if self.implementation == Implementation.hpat.value:
+            return hpat.jit(self.func)
+        elif self.implementation == Implementation.njit.value:
+            return numba.njit(self.func)
+        elif self.implementation == Implementation.native.value:
+            return self.func
+
+        raise ValueError(f'Unknown implementation: {self.implementation}')
+
+    def run(self, *args, **kwargs):
+        return self.runner(*args, **kwargs)

tests_perf/unicode_strings.py

@@ -0,0 +1,27 @@
+from pandas.util import testing as tm
+
+from .common import Implementation, ImplRunner
+
+
+class Methods:
+    params = [Implementation.native.value, Implementation.njit.value]
+    param_names = ['implementation']
+
+    def setup(self, implementation):
+        self.s = tm.randu(10 ** 4)
+        self.width = 10 ** 8
+
+    def time_center(self, implementation):
+        def _center(s, width):
+            return s.center(width)
+        ImplRunner(implementation, _center).run(self.s, self.width)
+
+    def time_ljust(self, implementation):
+        def _ljust(s, width):
+            return s.ljust(width)
+        ImplRunner(implementation, _ljust).run(self.s, self.width)
+
+    def time_rjust(self, implementation):
+        def _rjust(s, width):
+            return s.ljust(width)
+        ImplRunner(implementation, _rjust).run(self.s, self.width)