Jerry
70 lines
2.7 KiB
Python
70 lines
2.7 KiB
Python
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from cadpy import generation, step_artifacts
|
|
from cadpy.step_targets import ResolvedStepTarget
|
|
|
|
|
|
class StepArtifactsTests(unittest.TestCase):
|
|
def test_existing_step_target_ignores_python_source_for_glb_regeneration(self) -> None:
|
|
with tempfile.TemporaryDirectory() as temporary_directory:
|
|
root = Path(temporary_directory)
|
|
step_path = root / "part.step"
|
|
source_path = root / "part.py"
|
|
step_path.write_text("ISO-10303-21;\nEND-ISO-10303-21;\n", encoding="utf-8")
|
|
source_path.write_text("def gen_step():\n return None\n", encoding="utf-8")
|
|
|
|
target = ResolvedStepTarget(
|
|
cad_path="part",
|
|
kind="part",
|
|
source_path=source_path,
|
|
step_path=step_path,
|
|
)
|
|
self.assertIsNone(step_artifacts._python_source_for_target(target))
|
|
|
|
def test_missing_logical_step_can_still_use_python_source(self) -> None:
|
|
with tempfile.TemporaryDirectory() as temporary_directory:
|
|
root = Path(temporary_directory)
|
|
step_path = root / "part.step"
|
|
source_path = root / "part.py"
|
|
source_path.write_text("def gen_step():\n return None\n", encoding="utf-8")
|
|
|
|
target = ResolvedStepTarget(
|
|
cad_path="part",
|
|
kind="part",
|
|
source_path=step_path,
|
|
step_path=step_path,
|
|
)
|
|
self.assertEqual(step_artifacts._python_source_for_target(target), source_path)
|
|
|
|
def test_existing_step_spec_can_reuse_python_backed_glb_when_step_hash_matches(self) -> None:
|
|
with tempfile.TemporaryDirectory() as temporary_directory:
|
|
root = Path(temporary_directory)
|
|
step_path = root / "part.step"
|
|
step_path.write_text("ISO-10303-21;\nEND-ISO-10303-21;\n", encoding="utf-8")
|
|
|
|
spec = generation.EntrySpec(
|
|
source_ref="part.step",
|
|
cad_ref="part",
|
|
kind="part",
|
|
source_path=step_path,
|
|
display_name="part",
|
|
source="imported",
|
|
step_path=step_path,
|
|
)
|
|
self.assertFalse(generation._artifact_source_kind_matches_spec(spec, {"sourceKind": "python"}))
|
|
self.assertTrue(
|
|
generation._artifact_source_kind_matches_spec(
|
|
spec,
|
|
{
|
|
"sourceKind": "python",
|
|
"stepHash": generation.step_file_hash(step_path),
|
|
},
|
|
)
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|