From 4a10b7eb527a920841e5d80208dafeb0e328db9e Mon Sep 17 00:00:00 2001 From: narugo1992 Date: Mon, 9 Dec 2024 00:26:37 +0800 Subject: [PATCH] dev(narugo): fix the dwpose unittest --- test/pose/test_dwpose.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/test/pose/test_dwpose.py b/test/pose/test_dwpose.py index ffd75e00f7..dc3126e3aa 100644 --- a/test/pose/test_dwpose.py +++ b/test/pose/test_dwpose.py @@ -29,8 +29,12 @@ def test_dwpose_estimate_2girls(self, img_file_2girls, pose_data_2girls_0, pose_ assert len(poses) == 2 assert isinstance(poses[0], OP18KeyPointSet) assert isinstance(poses[1], OP18KeyPointSet) - assert np.isclose(pose_data_2girls_0, poses[0].all, atol=1).all() - assert np.isclose(pose_data_2girls_1, poses[1].all, atol=1).all() + c0 = np.isclose(pose_data_2girls_0, poses[0].all, atol=1) + assert c0.all(), f'{c0.mean() * 100:.2f}% of the values are the same, ' \ + f'max diff is {(pose_data_2girls_0 - poses[0].all).max():.4f}.' + c1 = np.isclose(pose_data_2girls_1, poses[1].all, atol=1) + assert c1.all(), f'{c1.mean() * 100:.2f}% of the values are the same, ' \ + f'max diff is {(pose_data_2girls_1 - poses[1].all).max():.4f}.' def test_dwpose_estimate_nothing(self, img_file_gun): poses = dwpose_estimate(img_file_gun)