update files and add tests

.vscode/settings.json

@@ -31,5 +31,8 @@
   ],
   "clangd.path": "clangd-17",
   "jupyter.debugJustMyCode": false,
-  "debugpy.debugJustMyCode": false
+  "debugpy.debugJustMyCode": false,
+  "pylint.args": ["--prefer-stubs=true"],
+  "python.testing.pytestEnabled": true,
+  "python.testing.unittestEnabled": false
 }

notebooks/example.ipynb

@@ -35,6 +35,46 @@
     "\n",
     "print(util.is_namespace(\"myproj\"))"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import json\n",
+    "from io import StringIO\n",
+    "\n",
+    "import pylint.lint\n",
+    "from pylint.reporters import JSONReporter\n",
+    "\n",
+    "# Adjust PYTHONPATH\n",
+    "# sys.path.append('/datafs/users/dingxf/neutrino-physics-tutorial/scikit-build-cpp-python')\n",
+    "\n",
+    "# Define the Pylint options\n",
+    "pylint_options = [\n",
+    "    \"--reports=n\",\n",
+    "    \"--output-format=json\",\n",
+    "    \"--extension-pkg-allow-list=simulation._csrc._hello\",\n",
+    "    \"--prefer-stubs=true\",\n",
+    "    \"--clear-cache-post-run=y\",\n",
+    "    \"/datafs/users/dingxf/neutrino-physics-tutorial/scikit-build-cpp-python/python/myproj/simulation/core.py\",  # The file you want to lint\n",
+    "]\n",
+    "\n",
+    "# Create a StringIO buffer to capture Pylint output\n",
+    "pylint_output = StringIO()\n",
+    "reporter = JSONReporter(output=pylint_output)\n",
+    "\n",
+    "# Run Pylint\n",
+    "pylint.lint.Run(pylint_options, reporter=reporter, exit=True)\n",
+    "\n",
+    "# Get the output\n",
+    "output = pylint_output.getvalue()\n",
+    "\n",
+    "# Parse and display the results\n",
+    "results = json.loads(output)\n",
+    "print(json.dumps(results, indent=2))"
+   ]
   }
  ],
  "metadata": {

python/myproj/simulation/core.py

+from ._csrc._hello import add, complexf
+
+
 def div(a: float, b: float) -> float:
     return a / b
+
+
+def complex2(a: float, b: float) -> float:
+    return complexf(a, b) + add(a, b)

python/tests/test_simulation.py

+#############################################################################
+#  Author: Xuefeng Ding <dingxf@ihep.ac.cn> @ IHEP-CAS
+#
+#  Project: cpp-python-small
+#  Date: 2024 November 17th
+#  Version: v1.0
+#  Description:
+#    Boilerplate for c++-python project.
+#    Can be installed with `pip install .`
+#
+#  Maintainer:
+#    Xuefeng Ding <dingxf@ihep.ac.cn>
+#
+#  All rights reserved. 2024 copyrighted.
+#############################################################################
+import pytest
+from myproj.simulation import add, complexf
+
+
+@pytest.mark.parametrize(
+    "a, b, expected",
+    [
+        (1, 2, 3),
+        (10, 15, 25),
+        (-5, 5, 0),
+        (0, 0, 0),
+    ],
+)
+def test_add_parametrized(a: float, b: float, expected: float) -> None:
+    assert add(a, b) == expected
+
+
+@pytest.mark.parametrize(
+    "a, b, expected",
+    [
+        (0, 1, 3),
+        (10, -5, 7),
+        (-10, -10, -18),
+    ],
+)
+def test_myclass_parametrized(a: float, b: float, expected: float) -> None:
+    assert complexf(a, b) == expected