Parallelize pytests

AI-multimodal · Oct 8, 2023 · 0bbb05d · 0bbb05d
1 parent cc65b25
commit 0bbb05d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -38,7 +38,7 @@ jobs:
 
     - name: Run tests
       run: |
-        pytest -v --cov --cov-report xml lightshow/_tests
+        pytest -n auto -v --cov --cov-report xml lightshow/_tests
 
     - name: Upload code coverage
       uses: codecov/codecov-action@v2

diff --git a/lightshow/_tests/conftest.py b/lightshow/_tests/conftest.py
@@ -64,7 +64,7 @@ def _database_for_stress_test():
     # return Database.from_materials_project(material_ids=["mp-980945"])
     db = Database.from_materials_project(chemsys=["Ti-*", "Mn-O-*"])
     # db = Database.from_materials_project(material_ids=["mp-390"])
-    keys = list(db._structures.keys())
+    keys = sorted(list(db._structures.keys()))
     # random.seed(123)
     # keys = random.sample(keys, 200)
     db._structures = {key: db._structures[key] for key in keys}

diff --git a/pyproject.toml b/pyproject.toml
@@ -56,6 +56,7 @@ test = [
     "coverage",
     "flake8",
     "pytest",
+    "pytest-xdist==3.3.1",
     "pytest-cov",
     "black",
     "nbstripout",