-
Notifications
You must be signed in to change notification settings - Fork 23
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
conftest update to supply unknown-format fake poses
- Loading branch information
Showing
1 changed file
with
16 additions
and
4 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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) |