优化流程

cadfs_to_cdsl/tests/test_parser.py

@@ -0,0 +1,47 @@
+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()