Supprimer/remplacer l'importation

Question

J'essaie de configurer un script de notation pour une classe CS intro en utilisant unittest. Essentiellement, les étudiants soumettent un fichier python student.py, qui a un certain nombre de fonctions qui sont généralement interdépendantes (ce qui signifie que func3() peut utiliser func1() dans ses calculs).

J'écris des tests unitaires pour chaque méthode en comparant la sortie de student.func1 à la sortie de correct.func1, une méthode qui est connue pour être la mise en œuvre correcte (à partir d'un fichier correct.py).

À titre d'exemple, disons que func2 utilise func1 dans son calcul. Donc, soit par défaut ou sur student.func1 échouant à un test, je veux écraser student.func1 avec correct.func1, alors student.func2 utilise une implémentation correcte connue (et n'est donc pas seulement faux par défaut). Comment pourrais-je faire ça? Il semble que les setUp() et tearDown() sont similaires à ce que je veux, mais je ne sais pas comment "désimporter" des modules en python, et je n'ai trouvé aucune ressource à ce sujet.

Je suis intéressé à la fois le cas où student.py contient des classes et func1, func2 sont des méthodes d'une classe spécifique, et quand func1 et func2 sont définis génériquement juste dans les student.py.

Answer 1

La façon la plus simple serait d'importer student dans votre module, puis prendre un AssertionError si un test échoue, et remplacer le mauvais code à l'intérieur du module étudiant avec votre propre bon code:

import student 

import unittest 

def safe_f1(): 
    print("Safe f1") 
    return 1 

class TestSomething(unittest.TestCase): 

    def test_f1(self): 
     try: 
      self.assertEqual(student.func1(), 1) 
     except AssertionError: 
      student.func1 = safe_f1 
      raise 


    def test_f2(self): 
     self.assertEqual(student.func2(), 2) 

est ici un mannequin student.py qui ne/fonctionne:

def func1(): 
    print("Bad f1") 
    return 2 

def func2(): 
    return func1() + 1 
    return 2 

Quand je cours, j'obtiens:

$ python -m unittest test.py 
Bad f1 
FSafe f1 
. 
====================================================================== 
FAIL: test_f1 (test.TestSomething) 
---------------------------------------------------------------------- 
Traceback (most recent call last): 
    File "/Users/austin/Code/so/test.py", line 13, in test_f1 
    self.assertEqual(student.func1(), 1) 
AssertionError: 2 != 1 

---------------------------------------------------------------------- 
Ran 2 tests in 0.001s 

FAILED (failures=1)