ci: add testing

Signed-off-by: Henry Schreiner <henryschreineriii@gmail.com>

Author: henryiii

.github/workflows/ci.yml

@@ -38,6 +38,20 @@ jobs:
     - uses: wntrblm/nox@2024.04.15
     - run: nox -s ${{ matrix.session }}
 
+  free-threading:
+    runs-on: ${{ matrix.runs-on }}
+    strategy:
+      fail-fast: false
+      matrix:
+        runs-on: [ubuntu-latest, windows-latest]
+    name: cibw on ${{ matrix.runs-on }}
+    steps:
+    - uses: actions/checkout@v4
+    - uses: pypa/cibuildwheel@v2.18
+      working-directory: projects/hello-free-threading
+      env:
+        CIBW_PRERELEASE_PYTHONS: True
+
   pass:
     if: always()
     needs: [checks]

.gitignore

@@ -56,6 +56,8 @@ docs/_build
 # build output (testing)
 skbuild/cmake_test_compile/*
 *.env
+wheelhouse
+dist/*
 
 # Generated manifests
 MANIFEST

noxfile.py

@@ -15,6 +15,8 @@
     "core-pybind11-hello",
     "hatchling-pybind11-hello",
 ]
+if sys.version_info >= (3, 13):
+    hello_list.append("hello-free-threading")
 
 
 @nox.session

projects/hello-free-threading/pyproject.toml

@@ -5,3 +5,9 @@ build-backend = "scikit_build_core.build"
 [project]
 name = "freecomputepi"
 version = "0.0.1"
+
+[tool.cibuildwheel]
+build = "cp313*"
+free-threaded-support = true
+test-command = "pytest {project} --durations=0"
+test-requires = ["pytest"]

projects/hello-free-threading/src/freecomputepi/comp.py

@@ -5,24 +5,25 @@
 
 from ._core import pi
 
-num_trials = 10_000_000
+
+def pi_in_threads(threads: int, trials: int) -> float:
+    if threads == 0:
+        return pi(trials)
+    with ThreadPoolExecutor(max_workers=threads) as executor:
+        return statistics.mean(executor.map(pi, [trials // threads] * threads))
 
 
-def main():
+def main() -> None:
     parser = argparse.ArgumentParser()
     parser.add_argument("--threads", type=int, default=0)
+    parser.add_argument("--trials", type=int, default=100_000_000)
     args = parser.parse_args()
-    threads = args.threads
-    start = time.monotonic()
-
-    if threads == 0:
-        π = pi(num_trials)
-    else:
-        with ThreadPoolExecutor(max_workers=threads) as executor:
-            π = statistics.mean(executor.map(pi, [num_trials // threads] * threads))
 
+    start = time.monotonic()
+    π = pi_in_threads(args.threads, args.trials)
     stop = time.monotonic()
-    print(f"{num_trials} trials, {π = }, {stop - start:.4} s")
+
+    print(f"{args.trials} trials, {π = }, {stop - start:.4} s")
 
 
 if __name__ == "__main__":

projects/hello-free-threading/src/freecomputepi/pure.py

@@ -4,39 +4,40 @@
 import time
 from concurrent.futures import ThreadPoolExecutor
 
-num_trials = 10_000_000
 
-
-def π(num_trials: int) -> float:
+def pi(trials: int) -> float:
     Ncirc = 0
     ran = random.Random()
 
-    for _ in range(num_trials):
+    for _ in range(trials):
         x = ran.uniform(-1, 1)
         y = ran.uniform(-1, 1)
 
         test = x * x + y * y
         if test <= 1:
             Ncirc += 1
 
-    return 4.0 * (Ncirc / num_trials)
+    return 4.0 * (Ncirc / trials)
+
 
+def pi_in_threads(threads: int, trials: int) -> float:
+    if threads == 0:
+        return pi(trials)
+    with ThreadPoolExecutor(max_workers=threads) as executor:
+        return statistics.mean(executor.map(pi, [trials // threads] * threads))
 
-def main():
+
+def main() -> None:
     parser = argparse.ArgumentParser()
     parser.add_argument("--threads", type=int, default=0)
+    parser.add_argument("--trials", type=int, default=10_000_000)
     args = parser.parse_args()
-    threads = args.threads
-    start = time.monotonic()
-
-    if threads == 0:
-        pi = π(num_trials)
-    else:
-        with ThreadPoolExecutor(max_workers=threads) as executor:
-            pi = statistics.mean(executor.map(π, [num_trials // threads] * threads))
 
+    start = time.monotonic()
+    π = pi_in_threads(args.threads, args.trials)
     stop = time.monotonic()
-    print(f"{num_trials} trials, pi is {pi}, {stop - start:.4} s")
+
+    print(f"{args.trials} trials, {π = }, {stop - start:.4} s")
 
 
 if __name__ == "__main__":

projects/hello-free-threading/tests/test_thread.py

@@ -0,0 +1,15 @@
+import freecomputepi.comp
+import freecomputepi.pure
+import pytest
+
+
+@pytest.mark.parametrize("threads", [0, 1, 2, 4])
+def test_pure(threads):
+    π = freecomputepi.pure.pi_in_threads(threads, 10_000_000)
+    assert π == pytest.approx(3.1415926535, rel=0.01)
+
+
+@pytest.mark.parametrize("threads", [0, 1, 2, 4])
+def test_comp(threads):
+    π = freecomputepi.comp.pi_in_threads(threads, 100_000_000)
+    assert π == pytest.approx(3.1415926535, rel=0.01)