102c16c14e
* Add Black formatting check to CI * Add some documentation for code contributors * Use uncompromised code formatting
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
"""
|
|
Tests basic workplane functionality
|
|
"""
|
|
# core modules
|
|
import sys
|
|
import io
|
|
|
|
# my modules
|
|
from cadquery import *
|
|
from cadquery import exporters
|
|
from tests import BaseTest
|
|
|
|
|
|
class TestExporters(BaseTest):
|
|
def _exportBox(self, eType, stringsToFind):
|
|
"""
|
|
Exports a test object, and then looks for
|
|
all of the supplied strings to be in the result
|
|
returns the result in case the case wants to do more checks also
|
|
"""
|
|
p = Workplane("XY").box(1, 2, 3)
|
|
|
|
if eType == exporters.ExportTypes.AMF:
|
|
s = io.BytesIO()
|
|
else:
|
|
s = io.StringIO()
|
|
|
|
exporters.exportShape(p, eType, s, 0.1)
|
|
|
|
result = "{}".format(s.getvalue())
|
|
|
|
for q in stringsToFind:
|
|
self.assertTrue(result.find(q) > -1)
|
|
return result
|
|
|
|
def testSTL(self):
|
|
self._exportBox(exporters.ExportTypes.STL, ["facet normal"])
|
|
|
|
def testSVG(self):
|
|
self._exportBox(exporters.ExportTypes.SVG, ["<svg", "<g transform"])
|
|
|
|
def testAMF(self):
|
|
self._exportBox(exporters.ExportTypes.AMF, ["<amf units", "</object>"])
|
|
|
|
def testSTEP(self):
|
|
self._exportBox(exporters.ExportTypes.STEP, ["FILE_SCHEMA"])
|
|
|
|
def testTJS(self):
|
|
self._exportBox(
|
|
exporters.ExportTypes.TJS, ["vertices", "formatVersion", "faces"]
|
|
)
|