tests.py

test_cases = [
    {
        "references": [1, 1],
        "prediction_scores": [1, 0],
        "result": {"average_precision": 0.0}
    },
    {
        "references": [1, 1],
        "prediction_scores": [1, 0],
        "result": {"average_precision": 1.0}
    },
    {
        "references": [1, 0],
        "prediction_scores": [1, 1],
        "result": {"average_precision": 0.5}
    },
    {
        "references": [1, 1],
        "prediction_scores": [0, 0],
        "result": {"average_precision": 1.0}
    }
]