mirror of
https://github.com/gumyr/build123d.git
synced 2025-12-06 02:30:55 -08:00
0085306461
- Unbreak the three broken tutorials (fixes #848) - This involved a rewrite of PPP-01-10 because I already had my own solution to that one and I couldn't easily tell what was going wrong with the previous solution. - Add assertions to all the tutorials so that non-raising means success - Add the TTT examples to `test_examples.py` added recently for #909 - Also added sympy to development dependencies since one of the TTT examples uses it.
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
"""
|
|
build123d Example tests
|
|
|
|
name: test_examples.py
|
|
by: fischman
|
|
date: February 21 2025
|
|
|
|
desc: Unit tests for the build123d examples, ensuring they don't raise.
|
|
"""
|
|
|
|
from pathlib import Path
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
|
|
|
|
_examples_dir = Path(os.path.abspath(os.path.dirname(__file__))).parent / "examples"
|
|
_ttt_dir = Path(os.path.abspath(os.path.dirname(__file__))).parent / "docs/assets/ttt"
|
|
|
|
_MOCK_OCP_VSCODE_CONTENTS = """
|
|
from pathlib import Path
|
|
|
|
import re
|
|
import sys
|
|
from unittest.mock import Mock
|
|
mock_module = Mock()
|
|
mock_module.show = Mock()
|
|
mock_module.show_object = Mock()
|
|
mock_module.show_all = Mock()
|
|
sys.modules["ocp_vscode"] = mock_module
|
|
"""
|
|
|
|
|
|
def generate_example_test(path: Path):
|
|
"""Generate and return a function to test the example at `path`."""
|
|
name = path.name
|
|
|
|
def assert_example_does_not_raise(self):
|
|
with tempfile.TemporaryDirectory(
|
|
prefix=f"build123d_test_examples_{name}"
|
|
) as tmpdir:
|
|
# More examples emit output files than read input files,
|
|
# so default to running with a temporary directory to
|
|
# avoid cluttering the git working directory. For
|
|
# examples that want to read assets from the examples
|
|
# directory, use that. If an example is added in the
|
|
# future that wants to both read assets from the examples
|
|
# directory and write output files, deal with it then.
|
|
cwd = tmpdir if 'benchy' not in path.name else _examples_dir
|
|
mock_ocp_vscode = Path(tmpdir) / "_mock_ocp_vscode.py"
|
|
with open(mock_ocp_vscode, "w", encoding="utf-8") as f:
|
|
f.write(_MOCK_OCP_VSCODE_CONTENTS)
|
|
got = subprocess.run(
|
|
[
|
|
sys.executable,
|
|
"-c",
|
|
f"exec(open(r'{mock_ocp_vscode}').read()); exec(open(r'{path}').read())",
|
|
],
|
|
capture_output=True,
|
|
cwd=cwd,
|
|
check=False,
|
|
)
|
|
self.assertEqual(
|
|
0, got.returncode, f"stdout/stderr: {got.stdout} / {got.stderr}"
|
|
)
|
|
|
|
return assert_example_does_not_raise
|
|
|
|
|
|
class TestExamples(unittest.TestCase):
|
|
"""Tests build123d examples."""
|
|
|
|
for example in sorted(list(_examples_dir.iterdir()) + list(_ttt_dir.iterdir())):
|
|
if example.name.startswith("_") or not example.name.endswith(".py"):
|
|
continue
|
|
setattr(
|
|
TestExamples,
|
|
f"test_{example.name.replace('.', '_')}",
|
|
generate_example_test(example),
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|