add profile run to ci

.github/workflows/ci.yml

@@ -125,3 +125,14 @@ jobs:
       default_python: '3.11'
       envs: |
         - linux: compatibility
+
+  profile:
+    needs: [core]
+    uses: OpenAstronomy/github-actions-workflows/.github/workflows/tox.yml@v1
+    with:
+      submodules: false
+      # Any env name which does not start with `pyXY` will use this Python version.
+      default_python: '3.12'
+      envs: |
+        - linux: profile
+          artifact-path: .profiles/*

asdf/conftest.py

@@ -1,4 +1,5 @@
 import os
+from pathlib import Path
 
 import pytest
 
@@ -8,6 +9,29 @@
 collect_ignore = ["asdf.py", "stream.py"]
 
 
+try:
+    from pyinstrument import Profiler
+
+    TESTS_ROOT = Path.cwd()
+
+    @pytest.fixture(autouse=True)
+    def auto_profile(request):
+        PROFILE_ROOT = TESTS_ROOT / ".profiles"
+        # Turn profiling on
+        profiler = Profiler()
+        profiler.start()
+
+        yield  # Run test
+
+        profiler.stop()
+        PROFILE_ROOT.mkdir(exist_ok=True)
+        results_file = PROFILE_ROOT / f"{request.node.name}.html"
+        profiler.write_html(results_file)
+
+except ImportError:
+    pass
+
+
 def pytest_collection_modifyitems(items):
     # Turn warnings into errors for all tests, this is needed
     # as running tests through pyargs will not use settings

tox.ini

@@ -13,6 +13,7 @@ env_list =
     weldx
     sunpy
     dkist
+    profile
 
 [testenv]
 set_env =
@@ -324,3 +325,10 @@ commands =
 # https://github.com/sunpy/sunpy/pull/7432
     pytest dkist --benchmark-skip \
         -W "ignore::asdf.exceptions.AsdfManifestURIMismatchWarning"
+
+[testenv:profile]
+commands_pre =
+    pip freeze
+extras = pyinstrument
+commands =
+    pyinstrument -m pytest