tests.py

test_cases = [
    {
        "generations": ["[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]"],
        "golds": [1, 3],
        "n_chances": 10,
        "result": {'accuracy@1': 0.5, 'accuracy@3': 1.0, 'accuracy@5': 1.0, 'accuracy@10': 1.0}
    },
    {
        "generations": ["[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]"],
        "golds": [3, 5],
        "n_chances": 10,
        "result": {'accuracy@1': 0, 'accuracy@3': 0.5, 'accuracy@5': 1.0, 'accuracy@10': 1.0}
    }
]