Skip to content

Commit

Permalink
conftest update to supply unknown-format fake poses
Browse files Browse the repository at this point in the history
  • Loading branch information
cleong110 committed Jan 10, 2025
1 parent c224568 commit 996f6f3
Showing 1 changed file with 16 additions and 4 deletions.
20 changes: 16 additions & 4 deletions src/python/pose_format/utils/conftest.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,23 @@
import copy
from typing import List
from typing import List, get_args
import pytest
from pose_format import Pose
from pose_format.utils.generic import get_standard_components_for_known_format, fake_pose
from pose_format.utils.generic import get_standard_components_for_known_format, fake_pose, KnownPoseFormat

@pytest.fixture
def fake_poses(request) -> List[Pose]:
# Access the parameter passed to the fixture
known_format = request.param
components = get_standard_components_for_known_format(known_format)
return copy.deepcopy([fake_pose(i * 10 + 10, components=components) for i in range(3)])
count = getattr(request, "count", 3)
known_formats = get_args(KnownPoseFormat)
if known_format in known_formats:

components = get_standard_components_for_known_format(known_format)
return copy.deepcopy([fake_pose(i * 10 + 10, components=components) for i in range(count)])
else:
# get openpose
fake_poses_list = [fake_pose(i * 10 + 10) for i in range(count)]
for i, pose in enumerate(fake_poses_list):
for component in pose.header.components:
component.name = f"unknown_component_{i}_formerly_{component.name}"
return copy.deepcopy(fake_poses_list)

0 comments on commit 996f6f3

Please sign in to comment.