Merge pull request #29 from numpy/asv-improvements

Asv improvements

Author: ngoldbaum

.gitignore

@@ -132,3 +132,4 @@ dmypy.json
 compile_commands.json
 
 .ruff-cache/
+.asv

benchmarks/.gitignore renamed to asv_benchmarks/.gitignore

@@ -16,18 +16,18 @@
     // The Python project's subdirectory in your repo.  If missing or
     // the empty string, the project is assumed to be located at the root
     // of the repository.
-    "repo_subdir": "",
+    // "repo_subdir": "",
 
     // Customizable commands for building, installing, and
     // uninstalling the project. See asv.conf.json documentation.
     //
     "install_command": [
       "pip install -i https://pypi.anaconda.org/scipy-wheels-nightly/simple numpy",
       "pip install meson-python patchelf wheel",
-      "in-dir={conf_dir} pip install ./asciidtype/ --no-build-isolation",
-      "in-dir={conf_dir} pip install ./strptrdtype/ --no-build-isolation"
+      "in-dir={conf_dir} pip install ../asciidtype/ --no-build-isolation",
+      "in-dir={conf_dir} pip install ../stringdtype/ --no-build-isolation"
     ],
-    "uninstall_command": ["return-code=any pip uninstall -y asciidtype strptrdtype"],
+    "uninstall_command": ["return-code=any pip uninstall -y asciidtype stringdtype"],
     "build_command": [
       "pip -V"
         /* "python setup.py build", */
@@ -56,7 +56,7 @@
     //"install_timeout": 600,
 
     // the base URL to show a commit for the project.
-    // "show_commit_url": "http://github.com/owner/project/commit/",
+    "show_commit_url": "http://github.com/numpy/numpy-user-dtypes/commit/",
 
     // The Pythons you'd like to test against.  If not provided, defaults
     // to the current version of Python used to run `asv`.
@@ -82,15 +82,9 @@
     // new environments.  A value of ``null`` means that the variable
     // will not be set for the current combination.
     //
-    // "matrix": {
-    //     "req": {
-    //         "numpy": ["1.6", "1.7"],
-    //         "six": ["", null],  // test with and without six installed
-    //         "pip+emcee": [""]   // emcee is only available for install with pip.
-    //     },
-    //     "env": {"ENV_VAR_1": ["val1", "val2"]},
-    //     "env_nobuild": {"ENV_VAR_2": ["val3", null]},
-    // },
+    "matrix": {
+        "env_nobuild": {"NUMPY_EXPERIMENTAL_DTYPE_API": "1"},
+    },
 
     // Combinations of libraries/python versions can be excluded/included
     // from the set to test. Each entry is a dictionary containing additional

benchmarks/asv.conf.json renamed to asv_benchmarks/asv.conf.json

@@ -0,0 +1,49 @@
+# Write the benchmarking functions here.
+# See "Writing benchmarks" in the asv docs for more information.
+import uuid
+
+import numpy as np
+
+from asciidtype import ASCIIDType
+from stringdtype import StringDType
+
+
+def generate_data(n=100000):
+    """Generate data for the benchmarks.
+
+    The vast majority of the time spent benchmarking is generating data; generate it
+    once and store it to avoid having to do this every run.
+    """
+    strings_list = [str(uuid.uuid4()) + "\n" for i in range(n)]
+
+    with open("strings", "w") as f:
+        f.writelines(strings_list)
+
+
+class TimeASCIIDType:
+    def setup(self):
+        self.ascii_dtype_object = ASCIIDType(36)
+        with open("strings", "r") as f:
+            self.strings = f.readlines()
+
+    def time_allocate(self):
+        _ = np.array(self.strings, dtype=self.ascii_dtype_object)
+
+
+class TimeStringDType:
+    def setup(self):
+        self.string_dtype_object = StringDType()
+        with open("strings", "r") as f:
+            self.strings = f.readlines()
+
+    def time_allocate(self):
+        _ = np.array(self.strings, dtype=self.string_dtype_object)
+
+
+class TimeObjectDType:
+    def setup(self):
+        with open("strings", "r") as f:
+            self.strings = f.readlines()
+
+    def time_allocate(self):
+        _ = np.array(self.strings, dtype=object)