Utilisation du module unittest avec le calcul des nombres premiers

avant de démarrer cette partie, il est recommandé de consulter :

• le cours sur les tests unitaires
• Et le paragraphe sur mettre son programme sous forme de modules

Définition: Un test unitaire est un test réalisé sur une portion du programme, typiquement sur une fonction.

Le module unittest offre des outils de test de code, comme la classe TestCase. Le but est des vérifier que votre code génère des résultats corrects, conformes au attentes.

En pratique: test du calcul de la liste de nombres premiers

• Ouvrir l’editeur Trinket.io editeur python

• Dans le fichier main.py, on mettra les fonctions de test. Ce sera le fichier main que l’on executera.

• Créer un deuxieme fichier python, que l’on renommera prime.py. Celui-ci contiendra les fonctions à tester.

Contenu du fichier prime.py

def prem1(n) :
    """Renvoie True si l’entier est premier et False sinon"""
    # script à completer
    return True
    

def liste_prime1(N):
    """etablit une liste de nombres premiers de 0 a N"""
    L=[]
    for j in range(N+1):
        if prem1(j):
            L.append(j)
    return L

Contenu du fichier main.py

import unittest
import prime 

class PrimeTestCase(unittest.TestCase):
  """Test pour prime.py"""
  
  def test_prime(self):
    """compare avec liste de nombres premiers"""
    
    wiki =[2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97]
    
    premiers = prime.liste_prime1(100)
    self.assertEqual(premiers,wiki)
    
if __name__=='__main__':
  unittest.main()

Dans le fichier à executer, main.py: On commence par importer le module unittest ainsi que le module prime qui contient les fonctions à tester.

Les noms des classes de test se terminent généralement par TestCase (scenario de test). La classe créée pour le test, PrimeTestCase doit être une classe enfant de unittest.TestCase.

Chaque test à créer est une méthode. Toute méthode dont le nom commence par test_ est exécutée lors du test. La méthode test_prime1 appelle donc la fonction liste_prime du module prime avec la valeur 100, puis vérifie que son résultat est conforme à la liste wiki, grâce à un test d’assertion assertEqual.

Ici, la fonction prem1 n’est pas bien renseignée, et le test échoue. On peut remonter jusqu’à l’erreur dans le Traceback qui est affiché en console.

# extrait du Traceback
Tests failed in  PrimeTestCase.test_prime  
Ran 1 tests, passed: 0 failed: 1

A vous de jouer

Compléter la fonction prem1 afin que celle-ci calcule bien les nombres premiers entre 0 et 100. Puis lancer le test à l’aide de la classe PrimeTestCase.

• Le test se déroule t-il correctement?
• Que se passe t-il si on modifie la valeur 100 dans l’instruction premiers = prime.liste_prime1(100) de la méthode test_prime de la classe PrimeTestCase? Conclure

La suite du TP: La spirale d’Ulam

Lien vers la suite du TP