20 lines
611 B
Python
20 lines
611 B
Python
from agent.CADAgent import CADAgent
|
|
|
|
|
|
def test_cadagent_prompt_includes_artifact_tags() -> None:
|
|
prompt = CADAgent.chat_impl.__doc__ or ""
|
|
|
|
assert "<|code_file|>" in prompt
|
|
assert "<|output_file|>" in prompt
|
|
assert "Repeat the final `<|code_file|>` and `<|output_file|>` tags" in prompt
|
|
|
|
|
|
def test_cadagent_prompt_prefers_step_for_visual_feedback() -> None:
|
|
prompt = CADAgent.chat_impl.__doc__ or ""
|
|
|
|
assert (
|
|
"prefer passing the exported `.step`/`.stp` file into `get_visual_feedback`"
|
|
in prompt
|
|
)
|
|
assert "visual feedback should use the STEP/STP file" in prompt
|