Code profiling

pyproject.toml

@@ -14,9 +14,11 @@ dev = [
     "pre-commit>=4.2.0",
     "ruff>=0.11.4",
     "pytest>=7.0",
+    "pytest-profiling",
+    "flameprof",
 ]
 
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 python_files = "test_*.py"
-addopts = "-v"
+addopts = "-v -m 'not profiling'"

src/dementpy.py

@@ -103,4 +103,5 @@ def main():
     os.chdir('../'+output_folder)
     export(Output_init, site, outname)
 
-main()
+if __name__ == '__main__':
+    main()

tests/profiling/test_profiling.py

@@ -0,0 +1,18 @@
+"""Test(s) to use with pytest-profiling to identify bottlenecks in the code."""
+import pytest
+
+@pytest.mark.profiling
+def test_profiling(monkeypatch):
+
+    # Define the command line arguments
+    import sys
+    argv = ['dementpy.py', 'grassland', 'output', '20250402', 'scrubland']
+    monkeypatch.setattr(sys, 'argv', argv)
+
+    # Move to subfolder so input and output folders will be correct
+    import os
+    os.chdir('src')
+
+    # Run dementpy
+    import dementpy
+    dementpy.main()