Débusquer une erreur par les tests

Débusquer une erreur par les tests. On veut calculer la somme des $n$ premiers entiers impairs, $1+3+\cdots +(2n-1)$, qui vaut $n^2$ d'après le cours de mathématiques. Un élève propose :

def somme_impairs(n):
    S = 0
    for k in range(1, 2*n - 1, 2):
        S = S + k
    return S

• Écrire un test assert fondé sur la formule $n^2$, pour $n$ allant de $1$ à $49$, en une boucle.
• Le test échoue dès $n=1$. Calculer somme_impairs(n) pour $n$ de $1$ à $5$, localiser l'erreur et la corriger.