test_examples.py

importpytest

fromglobimportglob
fromitertoolsimportchain, count

frompathimportPath

fromdocutils.parsers.rstimportdirectives
fromdocutils.coreimportpublish_doctree
fromdocutils.utilsimportReporter

importcadqueryascq
fromcadqueryimportcqgi
fromcadquery.cq_directiveimportcq_directive


deffind_examples(pattern="examples/*.py", path=Path("examples")):

forpinglob(pattern):
withopen(p, encoding="UTF-8") asf:
code=f.read()

yieldcode, path


deffind_examples_in_docs(pattern="doc/*.rst", path=Path("doc")):

# dummy CQ directive for code
classdummy_cq_directive(cq_directive):

codes= []

defrun(self):

self.codes.append("\n".join(self.content))

return []

directives.register_directive("cadquery", dummy_cq_directive)

# read and parse all rst files
forpinglob(pattern):
withopen(p, encoding="UTF-8") asf:
doc=f.read()

publish_doctree(
doc, settings_overrides={"report_level": Reporter.SEVERE_LEVEL+1}
 )

# yield all code snippets
forcindummy_cq_directive.codes:

yieldc, path


@pytest.mark.parametrize(
"code, path", chain(find_examples(), find_examples_in_docs()), ids=count(0)
)
deftest_example(code, path):

# build
withpath:
res=cqgi.parse(code).build()

assertres.exceptionisNone

# check if the resulting objects are valid
forrinres.results:
r=r.shape
ifisinstance(r, cq.Workplane):
forvinr.vals():
ifisinstance(v, cq.Shape):
assertv.isValid()
elifisinstance(r, cq.Shape):
assertr.isValid()