48 lines
2.0 KiB
Python
48 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
from cadfs_to_cdsl.featurescript_lexer import lex
|
|
from cadfs_to_cdsl.featurescript_parser import parse_featurescript
|
|
from cadfs_to_cdsl.ir import Call
|
|
from cadfs_to_cdsl.query_parser import parse_query
|
|
from cadfs_to_cdsl.units import length_mm
|
|
|
|
|
|
SOURCE = r'''
|
|
FeatureScript 1511;
|
|
export const f = defineFeature(function(context, id, definition) {
|
|
{ var Q0; Q0=qCreatedBy(makeId("Top.planeOp"), FACE);
|
|
var sketch = newSketch(context, id + "F0", {"sketchPlane":qUnion([Q0])});
|
|
skCircle(sketch, "E0", {"center":v(0, 0) * mm, "radius":9.53 * mm}); skSolve(sketch); }
|
|
{ var Q0; Q0=qSketchRegion(id + "F0", true);
|
|
extrude(context, id + "F1", {"entities":qUnion([Q0]), "depth":120 * mm}); }
|
|
});
|
|
'''
|
|
|
|
|
|
class ParserTests(unittest.TestCase):
|
|
def test_lexer_ignores_comments_and_preserves_lines(self):
|
|
tokens = lex('// a\nfoo(/*b*/"x")')
|
|
self.assertEqual([token.value for token in tokens[:-1]], ["foo", "(", '"x"', ")"])
|
|
self.assertEqual(tokens[0].line, 2)
|
|
|
|
def test_nested_feature_script(self):
|
|
model = parse_featurescript(SOURCE, "00000173")
|
|
self.assertEqual([step.feature_id for step in model.steps], ["F0", "F1"])
|
|
self.assertEqual(model.sketches[0].entities[0].operation, "skCircle")
|
|
self.assertIsInstance(model.features[0].params["entities"], Call)
|
|
|
|
def test_query_parser(self):
|
|
query = Call("makeQuery", [Call("__binary__", ["id", "+", "F1.opExtrude"]), "CAP_EDGE", "EDGE", {"isStart": False, "x": Call("sQuery", [Call("__binary__", ["id", "+", "F0.wireOp"]), "EDGE", "E0"])}])
|
|
value = parse_query(query)
|
|
self.assertEqual((value.owner_feature, value.source_sketch, value.source_entity), ("F1", "F0", "E0"))
|
|
self.assertFalse(value.is_start)
|
|
|
|
def test_safe_units(self):
|
|
self.assertEqual(length_mm("2 * inch"), 50.8)
|
|
self.assertEqual(length_mm("25.4 / 2 * mm"), 12.7)
|
|
with self.assertRaises(ValueError): length_mm("external.value * mm")
|
|
|
|
|
|
if __name__ == "__main__": unittest.main()
|