U-C4N
88a9005a6a
Introduces an engineering/ package that lets MCP clients produce production- quality technical drawings without LLM-improvised geometry. Adds 7 new MCP tools, an 8-step pre-completion validator, and an automatic linetype + layer bootstrap on drawing_new so CENTER/HIDDEN/PHANTOM and standard mechanical layers (GEOMETRY, DIM, HATCH, TEXT, TITLEBLOCK, ...) are always available. - engineering/gear.py: pure-Python involute math, 40+ points/flank, helical + spur front views, deterministic section A-A (no helix lines in section). - engineering/keyway.py: DIN 6885 keyway table + parametric keyed bore / side-section drawers. - engineering/titleblock.py: ISO 7200 A3 (420x297 mm) title block; title text is written verbatim to the title cell. - engineering/validator.py: DrawingValidator with 8 checks (real DIMENSION entities, no hidden orphans, linetypes loaded, bore + keyway in both views when expected, title consistency e.g. forbid 'HELICAL SPUR GEAR' when helix_angle truthy, saved-to-disk + file-on-disk, screenshot exported, dwg path returned). - backends/ezdxf_backend.py: AutoCAD-only linetype fallback (HIDDEN, BORDER families) so ezdxf doc gets correct patterns when linetype_load is called for names ezdxf.tools.standards does not ship. - backends/base.py: concrete helpers set_layer_active / ensure_linetypes. - server.py: Section 12 with 7 engineering tools, drawing_new(bootstrap= True), and an ENGINEERING WORKFLOW block in the MCP instructions string. - 30 new tests covering gear math, DIN6885 lookup, ISO A3 layout, title verbatim, and every validator check. Tool count 89 -> 96.
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
"""Tests for engineering.keyway — DIN 6885 lookup + drawing helpers."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from engineering.keyway import (
|
|
draw_keyed_bore,
|
|
draw_keyway_section,
|
|
keyway_dimensions,
|
|
)
|
|
from engineering.layers import (
|
|
ensure_engineering_layers,
|
|
ensure_standard_linetypes,
|
|
)
|
|
|
|
|
|
def test_din6885_lookup_22_30_range():
|
|
dims = keyway_dimensions(25.0)
|
|
assert dims["width"] == 8.0
|
|
assert dims["height"] == 7.0
|
|
assert dims["depth_shaft"] == 4.0
|
|
|
|
|
|
def test_din6885_lookup_38_44_range():
|
|
dims = keyway_dimensions(40.0)
|
|
assert dims["width"] == 12.0
|
|
|
|
|
|
def test_keyway_dimensions_below_min_raises_or_clamps():
|
|
with pytest.raises(ValueError):
|
|
keyway_dimensions(3.0)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_draw_keyed_bore_returns_expected_handles(backend):
|
|
await ensure_standard_linetypes(backend)
|
|
await ensure_engineering_layers(backend)
|
|
res = await draw_keyed_bore(
|
|
backend, center=(0.0, 0.0), bore_diameter=25.0,
|
|
keyway_width=8.0, keyway_depth=4.0,
|
|
)
|
|
for k in ("bore", "keyway_polyline", "centerline_h", "centerline_v"):
|
|
assert k in res
|
|
assert res[k]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_draw_keyway_section_geometry(backend):
|
|
await ensure_standard_linetypes(backend)
|
|
await ensure_engineering_layers(backend)
|
|
res = await draw_keyway_section(
|
|
backend, center=(0.0, 0.0), bore_diameter=25.0, face_width=40.0,
|
|
keyway_width=8.0, keyway_depth=4.0,
|
|
)
|
|
assert "bore_top" in res
|
|
assert "bore_bottom" in res
|
|
assert isinstance(res["keyway"], list)
|
|
assert len(res["keyway"]) == 4
|
|
# Verify the notch top sits above the bore top by reading entities back.
|
|
bore_top = await backend.entity_get(res["bore_top"])
|
|
bore_top_y = bore_top.properties["start"][1]
|
|
notch_top_handle = res["keyway"][2] # notch_top in our return order
|
|
notch_top_ent = await backend.entity_get(notch_top_handle)
|
|
notch_top_y = notch_top_ent.properties["start"][1]
|
|
assert notch_top_y > bore_top_y
|