Use pytest for simpler tests

.devcontainer/.dockerfilelintrc

@@ -0,0 +1,2 @@
+rules:
+  sudo_usage: off

.devcontainer/Dockerfile

@@ -0,0 +1,24 @@
+FROM python:3.12-slim
+
+# Install essential packages and create non-root user
+RUN apt-get update && apt-get install -y --no-install-recommends git curl sudo bash-completion vim \
+    && useradd -m -s /bin/bash vscode \
+    && echo "vscode ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers \
+    && mkdir -p /workspace \
+    && chown vscode:vscode /workspace \
+    && rm -rf /var/lib/apt/lists/*
+
+WORKDIR /workspace
+
+# Switch to non-root user
+USER vscode
+
+# Set up shell completions
+RUN echo 'source /usr/share/bash-completion/completions/git' >> ~/.bashrc \
+    && echo 'source /etc/bash_completion' >> ~/.bashrc
+
+# Set up uv and Python environment
+RUN curl -LsSf https://astral.sh/uv/install.sh | sh
+
+# Set up environment variables
+ENV PATH="/home/vscode/.local/bin:${PATH}"

.devcontainer/devcontainer.json

@@ -0,0 +1,43 @@
+{
+    "name": "py-cppmodel Dev Container",
+    "build": {
+        "dockerfile": "Dockerfile",
+        "context": ".."
+    },
+    "customizations": {
+        "vscode": {
+            "extensions": [
+                "charliermarsh.ruff",
+                "esbenp.prettier-vscode",
+                "github.vscode-pull-request-github",
+                "ms-python.mypy-type-checker",
+                "ms-python.python",
+                "ms-python.vscode-pylance",
+                "ms-toolsai.jupyter-keymap",
+                "ms-toolsai.jupyter-renderers",
+                "ms-toolsai.jupyter",
+                "ms-toolsai.vscode-jupyter-cell-tags",
+                "ms-toolsai.vscode-jupyter-slideshow",
+            ],
+            "settings": {
+                "python.defaultInterpreterPath": "/${workspaceFolder}/.venv/bin/python",
+                "[python]": {
+                    "editor.defaultFormatter": "charliermarsh.ruff",
+                    "editor.formatOnSave": true,
+                    "editor.codeActionsOnSave": {
+                        "source.organizeImports": "explicit"
+                    }
+                }
+            }
+        }
+    },
+    "mounts": [
+        "source=${localWorkspaceFolder},target=/workspace,type=bind,consistency=cached",
+        "source=${localEnv:HOME}/.ssh,target=/home/vscode/.ssh,type=bind,consistency=cached"
+    ],
+    "remoteUser": "vscode",
+    "updateRemoteUserUID": true,
+    "remoteEnv": {
+        "UV_LINK_MODE": "copy"
+    }
+}

.github/workflows/python.yml

@@ -31,4 +31,4 @@ jobs:
       - name: Type checks
         run: uv run ty check .
       - name: Tests
-        run: uv run python -m unittest discover --verbose .
+        run: uv run pytest

LICENSE

@@ -1,6 +1,6 @@
 The MIT License (MIT)
 
-Copyright (c) 2016 The py_cppmodel Authors. All Rights Reserved.
+Copyright (c) 2016 The py-cppmodel Authors. All Rights Reserved.
 
 Permission is hereby granted, free of charge, to any person obtaining a copy of
 this software and associated documentation files (the "Software"), to deal in

README.md

@@ -25,7 +25,7 @@ uv sync
 To run the tests, run:
 
 ```sh
-uv run python -m unittest discover --verbose .
+uv run pytest
 ```
 
 To run type checking:

pyproject.toml

@@ -20,9 +20,14 @@ dev = [
     "jupyter>=1.1.1",
     "parameterized>=0.9.0",
     "pre-commit>=4.5.1",
+    "pytest>=9.0.2",
+    "pytest-xdist>=3.8.0",
     "ty>=0.0.14",
 ]
 
+[tool.pytest.ini_options]
+addopts = "-n auto -v"
+
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"

test.macos.sh

@@ -7,4 +7,4 @@ uv sync
 uv run ty check .
 
 # Unit tests
-uv run python -m unittest discover --verbose .
+uv run pytest

test_parse_standard_library_includes.py

@@ -1,7 +1,5 @@
-import unittest
-
+import pytest
 from clang.cindex import TranslationUnit
-from parameterized import parameterized
 
 import py_cppmodel
 
@@ -14,50 +12,41 @@
 ]
 
 
-def _custom_name_func(testcase_func, _, param):
-    return "%s_%s" % (testcase_func.__name__, parameterized.to_safe_name(param.args[0]))
-
-
-class TestStandardLibraryIncludes(unittest.TestCase):
-    @parameterized.expand(
-        [
-            "algorithm",
-            "any",
-            "array",
-            "deque",
-            "forward_list",
-            "functional",
-            "iterator",
-            "list",
-            "map",
-            "memory",
-            "numeric",
-            "optional",
-            "queue",
-            "set",
-            "stack",
-            "string",
-            "tuple",
-            "type_traits",
-            "unordered_map",
-            "unordered_set",
-            "utility",
-            "variant",
-            "vector",
-        ],
-        name_func=_custom_name_func,
+@pytest.mark.parametrize(
+    "include",
+    [
+        "algorithm",
+        "any",
+        "array",
+        "deque",
+        "forward_list",
+        "functional",
+        "iterator",
+        "list",
+        "map",
+        "memory",
+        "numeric",
+        "optional",
+        "queue",
+        "set",
+        "stack",
+        "string",
+        "tuple",
+        "type_traits",
+        "unordered_map",
+        "unordered_set",
+        "utility",
+        "variant",
+        "vector",
+    ],
+)
+def test_include(include):
+    source = f"#include <{include}>"
+    tu = TranslationUnit.from_source(
+        "t.cc",
+        COMPILER_ARGS,
+        unsaved_files=[("t.cc", source)],
     )
-    def test_include(self, include):
-        source = f"#include <{include}>"
-        tu = TranslationUnit.from_source(
-            "t.cc",
-            COMPILER_ARGS,
-            unsaved_files=[("t.cc", source)],
-        )
-
-        # This should not raise an exception.
-        self.model = py_cppmodel.Model(tu)
-
 
-if __name__ == "__main__":
-    unittest.main()
+    # This should not raise an exception.
+    py_cppmodel.Model(tu)

test_py_cppmodel.py

@@ -1,5 +1,4 @@
-import unittest
-
+import pytest
 from clang.cindex import TranslationUnit
 
 import py_cppmodel
@@ -35,59 +34,53 @@ class B {
 """
 
 
-class TestCppModel(unittest.TestCase):
-    def setUp(self):
-        tu = TranslationUnit.from_source(
-            "sample.cc",
-            COMPILER_ARGS,
-            unsaved_files=[("sample.cc", SOURCE)],
-        )
-        self.model = py_cppmodel.Model(tu)
-
-    def test_filename(self):
-        self.assertEqual(self.model.filename, "sample.cc")
-
-    def test_functions(self):
-        self.assertEqual(len(self.model.functions), 2)
-        self.assertEqual(str(self.model.functions[0]), "<py_cppmodel.Function double bar(double)>")
-        self.assertEqual(str(self.model.functions[1]), "<py_cppmodel.Function int main()>")
-
-    def test_classes(self):
-        self.assertEqual(len(self.model.classes), 1)
-        self.assertEqual(str(self.model.classes[0]), "<py_cppmodel.Class A>")
-
-        self.assertEqual(len(self.model.classes[0].annotations), 1)
-        self.assertEqual(self.model.classes[0].annotations[0], "A")
-
-        self.assertEqual(len(self.model.classes[0].members), 3)
-        self.assertEqual(
-            str(self.model.classes[0].members[0]),
-            "<py_cppmodel.Member <py_cppmodel.Type int> a>",
-        )
-        self.assertEqual(
-            str(self.model.classes[0].members[1]),
-            "<py_cppmodel.Member <py_cppmodel.Type double> b>",
-        )
-        self.assertEqual(
-            str(self.model.classes[0].members[2]),
-            "<py_cppmodel.Member <py_cppmodel.Type char[8]> c>",
-        )
-
-        self.assertEqual(len(self.model.classes[0].methods), 1)
-        self.assertEqual(str(self.model.classes[0].methods[0]), "<py_cppmodel.Method int foo(int)>")
-        self.assertEqual(len(self.model.classes[0].methods[0].annotations), 1)
-        self.assertEqual(self.model.classes[0].methods[0].annotations[0], "foo")
-
-        self.assertEqual(len(self.model.unmodelled_nodes), 2)
-        self.assertEqual(
-            str(self.model.unmodelled_nodes[0]),
-            "<py_cppmodel.Unmodelled z <SourceLocation file 'sample.cc', line 1, column 5>>",
-        )
-        self.assertEqual(
-            str(self.model.unmodelled_nodes[1]),
-            "<py_cppmodel.Unmodelled B<T> <SourceLocation file 'sample.cc', line 12, column 7>>",
-        )
-
-
-if __name__ == "__main__":
-    unittest.main()
+@pytest.fixture
+def model():
+    tu = TranslationUnit.from_source(
+        "sample.cc",
+        COMPILER_ARGS,
+        unsaved_files=[("sample.cc", SOURCE)],
+    )
+    return py_cppmodel.Model(tu)
+
+
+def test_filename(model):
+    assert model.filename == "sample.cc"
+
+
+def test_functions(model):
+    assert len(model.functions) == 2
+    assert str(model.functions[0]) == "<py_cppmodel.Function double bar(double)>"
+    assert str(model.functions[1]) == "<py_cppmodel.Function int main()>"
+
+
+def test_classes(model):
+    assert len(model.classes) == 1
+    assert str(model.classes[0]) == "<py_cppmodel.Class A>"
+
+    assert len(model.classes[0].annotations) == 1
+    assert model.classes[0].annotations[0] == "A"
+
+
+def test_class_members(model):
+    assert len(model.classes[0].members) == 3
+    assert str(model.classes[0].members[0]) == "<py_cppmodel.Member <py_cppmodel.Type int> a>"
+    assert str(model.classes[0].members[1]) == "<py_cppmodel.Member <py_cppmodel.Type double> b>"
+    assert str(model.classes[0].members[2]) == "<py_cppmodel.Member <py_cppmodel.Type char[8]> c>"
+
+    assert len(model.classes[0].methods) == 1
+    assert str(model.classes[0].methods[0]) == "<py_cppmodel.Method int foo(int)>"
+    assert len(model.classes[0].methods[0].annotations) == 1
+    assert model.classes[0].methods[0].annotations[0] == "foo"
+
+
+def test_unmodelled_nodes(model):
+    assert len(model.unmodelled_nodes) == 2
+    assert (
+        str(model.unmodelled_nodes[0])
+        == "<py_cppmodel.Unmodelled z <SourceLocation file 'sample.cc', line 1, column 5>>"
+    )
+    assert (
+        str(model.unmodelled_nodes[1])
+        == "<py_cppmodel.Unmodelled B<T> <SourceLocation file 'sample.cc', line 12, column 7>>"
+    )