tests.py

test_cases = [
    {
        "predictions": [0.01, 0.01, 0.9],
        "references": [2, 0],
    },
    {
        "predictions": [0.01, 0.01, 0.9],
        "references": [1, 2],
    },
    {
        "predictions": [0.9, 0.01, 0.01],
        "references": [0, 0],
    }
    {
        "predictions": [0.9, 0.01, 0.01],
        "references": [2, 2],
    }
    {
        "predictions": [0.01, 0.9, 0.01],
        "references": [1, 0],
    }
    {
        "predictions": [0.01, 0.9, 0.01],
        "references": [0, 2],
    }
]