Use pytest for simpler tests

Author: jbcoe

.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

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>>"
+    )