CedarandClaude Sonnet 4.6
80d982fa1d
Adds integration tests exercising the full ReorientFaceMesh -> RenderTexturedFaceMesh data flow with synthetic inputs, verifying output shape contracts and value ranges. Also adds pytest.ini and conftest.py for standalone test execution, guards __init__.py relative imports with try/except for pytest 8.x Package compatibility, and removes tests/ from .gitignore. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
37 lines
1.6 KiB
Python
37 lines
1.6 KiB
Python
try:
|
|
from .nodes.extract import ExtractSkeletonFromPose
|
|
from .nodes.transform import ComputeRetargetTransform, ApplyRetargetTransform
|
|
from .nodes.visualization import SkeletonToOpenPoseImage
|
|
from .nodes.face_landmarker import MediaPipeFaceLandmarker
|
|
from .nodes.reorient_face_mesh import ReorientFaceMesh
|
|
from .nodes.render_textured_face import RenderTexturedFaceMesh
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ExtractSkeletonFromPose": ExtractSkeletonFromPose,
|
|
"ComputeRetargetTransform": ComputeRetargetTransform,
|
|
"ApplyRetargetTransform": ApplyRetargetTransform,
|
|
"SkeletonToOpenPoseImage": SkeletonToOpenPoseImage,
|
|
"MediaPipeFaceLandmarker": MediaPipeFaceLandmarker,
|
|
"ReorientFaceMesh": ReorientFaceMesh,
|
|
"RenderTexturedFaceMesh": RenderTexturedFaceMesh,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ExtractSkeletonFromPose": "Extract Skeleton From Pose",
|
|
"ComputeRetargetTransform": "Compute Retarget Transform",
|
|
"ApplyRetargetTransform": "Apply Retarget Transform",
|
|
"SkeletonToOpenPoseImage": "Skeleton To OpenPose Image",
|
|
"MediaPipeFaceLandmarker": "MediaPipe Face Landmarker",
|
|
"ReorientFaceMesh": "Reorient Face Mesh",
|
|
"RenderTexturedFaceMesh": "Render Textured Face Mesh",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|
|
|
|
except ImportError:
|
|
# Running outside the ComfyUI package context (e.g. pytest standalone).
|
|
# Nodes are imported directly in tests via sys.path, so this is safe to skip.
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|