| 143 | |
| 144 | @pytest.mark.parametrize("inputs, expected_values", KEYPOINT_ERROR_TEST_DATA) |
| 145 | def test_evaluate_keypoint_error(inputs, expected_values): |
| 146 | keypoint_error = pet.keypoint_error( |
| 147 | inputs["df_error"], |
| 148 | inputs["df_error"], |
| 149 | inputs["train_indices"], |
| 150 | inputs["test_indices"], |
| 151 | ) |
| 152 | print(inputs["df_error"]) |
| 153 | print(keypoint_error) |
| 154 | for bodypart, mean_errors in expected_values.items(): |
| 155 | for error_name in KEYPOINT_ERROR_NAMES: |
| 156 | if "train" in error_name.lower(): |
| 157 | mean_error = mean_errors[0] |
| 158 | else: |
| 159 | mean_error = mean_errors[1] |
| 160 | |
| 161 | assert keypoint_error.loc[error_name, bodypart] == mean_error |
| 162 | |
| 163 | |
| 164 | def test_get_available_requested_snapshots_ok(): |