benchmark performance of import

Do a benchmark before working on #522

Signed-off-by: Jinzhe Zeng <[email protected]>

.github/workflows/test.yml

@@ -27,6 +27,11 @@ jobs:
       uses: codecov/codecov-action@v4
       env:
         CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
+    - name: Run benchmarks
+      uses: CodSpeedHQ/action@v2
+      with:
+        token: ${{ secrets.CODSPEED_TOKEN }}
+        run: pytest tests/benchmark/ --codspeed
   pass:
     needs: [build]
     runs-on: ubuntu-latest

pyproject.toml

@@ -58,6 +58,10 @@ docs = [
     'jupyterlite-sphinx',
     'jupyterlite-xeus-python',
 ]
+test = [
+    'pytest',
+    'pytest-codspeed',
+]
 
 [tool.setuptools.packages.find]
 include = ["dpdata*"]

tests/benchmark/test_import.py

@@ -0,0 +1,13 @@
+import pytest
+import subprocess
+import sys
+
+@pytest.mark.benchmark
+def test_import():
+    """Test import dpdata."""
+    subprocess.check_output([sys.executable, "-c", "'from dpdata import LabeledSystem'"]).decode("ascii")
+
+@pytest.mark.benchmark
+def test_cli():
+    """Test dpdata command."""
+    subprocess.check_output([sys.executable, "-m", "dpdata", "-h"]).decode("ascii")