fixes

Author: Joshix-1

examples/factoradic.py

@@ -41,7 +41,7 @@ def int_to_factoradic(num: int) -> str:
     return Stream(digits).map(DIGITS.__getitem__).collect("".join)
 
 
-def test() -> None:
+def test() -> int | str:
     """Test this."""
     test_cases: tuple[tuple[int, str], ...] = (
         (1, "1"),
@@ -59,8 +59,12 @@ def test() -> None:
     )
 
     for i, f in test_cases:
-        assert (_ := factoradic_to_int(f)) == i, f"{_!r} != {i!r}"
-        assert (_ := int_to_factoradic(i)) == f, f"{_!r} != {f!r}"
+        if (_ := factoradic_to_int(f)) != i:
+            return f"{_!r} != {i!r}"
+        if (_ := int_to_factoradic(i)) != f:
+            return f"{_!r} != {f!r}"
+
+    return 0
 
 
 if __name__ == "__main__":

examples/nth_prime.sh

@@ -1,10 +1,14 @@
 #!/bin/sh
+set -e
+
 if [ -z "$1" ]; then
   N="1"
 else
   N="$1"
 fi
 
+set -u
+
 if [ -f "./nth_prime.sh" ]; then
   DIR="."
 elif [ -f "./examples/nth_prime.sh" ]; then

examples/test/__main__.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from pathlib import Path
-from subprocess import PIPE, run
+from subprocess import PIPE, run  # nosec
 
 from ..factoradic import test
 from ..print_primes import test_prime_stream
@@ -16,9 +16,16 @@
 
 def _get_nth_prime(num: int) -> int:
     """Run nth_prime.sh."""
-    command = [(EXAMPLES_DIR / "nth_prime.sh").as_posix(), str(num)]
-    return int(run(command, check=True, stdout=PIPE).stdout)
-
-
-assert _get_nth_prime(100_000) == 1299709
-assert _get_nth_prime(200_000) == 2750159
+    return int(
+        run(  # nosec
+            ["./nth_prime.sh", str(num)],
+            check=True,
+            stdout=PIPE,
+            shell=False,
+            cwd=EXAMPLES_DIR,
+        ).stdout
+    )
+
+
+assert _get_nth_prime(100_000) == 1299709  # nosec
+assert _get_nth_prime(200_000) == 2750159  # nosec