Tout en étudiant autour de python unittest et pytest, si vous regardez la traduction japonaise officielle de la documentation unittest, Puisqu'il y avait des fonctions addModuleCleanup et doModuleCleanups ajoutées à partir de python3.8, je les présenterai. Qu'est-ce que unittest? Il peut être préférable d'avoir un aperçu de unittest dans un autre article que de lire cet article.
TL;DR
Vous pouvez spécifier la fonction à appeler lorsqu'une erreur se produit avec la fonction setUpModule.
Par exemple, essayez d'exécuter le code suivant. Ceci est un test pour confirmer l'addition et la soustraction. Pour vérifier quand la fonction est en cours d'exécution
print(sys._getframe().f_code.co_name)Le nom de la fonction est émis à.
```python
import sys
import unittest
# TODO(hatsumi)~ J'ajouterai le code plus tard ~
def createConnection():
print(sys._getframe().f_code.co_name)
# raise Exception
def closeConnection():
print(sys._getframe().f_code.co_name)
def setUpModule():
createConnection()
def tearDownModule():
closeConnection()
class TestOperator(unittest.TestCase):
@classmethod
def setUpClass(self):
print(sys._getframe().f_code.co_name)
@classmethod
def tearDownClass(self):
print(sys._getframe().f_code.co_name)
def setUp(self):
print(sys._getframe().f_code.co_name)
def tearDown(self):
print(sys._getframe().f_code.co_name)
def test_add(self):
print(sys._getframe().f_code.co_name)
self.assertEqual(2 + 3, 5)
def test_sub(self):
print(sys._getframe().f_code.co_name)
assert 3 - 2 == 1
if __name__ == "__main__":
unittest.main()
À la suite de l'exécution, les éléments suivants doivent être affichés.
createConnection
setUpClass
setUp
test_add
tearDown
.setUp
test_sub
tearDown
.tearDownClass
closeConnection
Vous pouvez voir que la configuration et le démontage sont appelés dans l'ordre de la méthode module-> classe->.
Si une exception se produit dans la fonction setUpModule, la fonction tearDownModule ne sera pas appelée. Au lieu de cela, la fonction ajoutée par la fonction addModuleCleanup est exécutée par la fonction doModuleCleanups.
Dans le code ci-dessus, la fonction setUpModule génère createConnection et la fonction tearDownModule renvoie closeConnection. Si vous décommentez la fonction setUpModule et exécutez `` lever Exception '', la fonction tearDownModule ne sera pas appelée. Cela peut être bon si vous gérez le traitement des connexions à la base de données et des choses temporaires en supposant que la fonction tearDownModule est appelée.
Afin d'exécuter la fonction, il est nécessaire d'enregistrer la fonction à l'avance. Pour ce faire, utilisez la fonction addModuleCleanup. Ajoutez le code suivant à la partie commentaire TODO du code ci-dessus. Enregistrez la fonction print_number.
def print_number():
print(sys._getframe().f_code.co_name)
unittest.addModuleCleanup(print_number)
Lorsqu'une exception se produit à la suite de l'exécution, la fonction print_number est appelée. La chaîne de caractères est également affichée dans createConnection. ..
print_number
createConnection
E
======================================================================
ERROR: setUpModule (__main__)
----------------------------------------------------------------------
Traceback (most recent call last):
File "test_module.py", line 25, in setUpModule
createConnection()
File "test_module.py", line 19, in createConnection
raise Exception
Exception
----------------------------------------------------------------------
Ran 0 tests in 0.001s
Il est également possible d'enregistrer plusieurs fonctions. Cependant, dans ce cas, il sera exécuté par LIFO. Ajoutez le code suivant à la partie commentaire TODO du code ci-dessus.
unittest.addModuleCleanup((lambda : print('a')))
unittest.addModuleCleanup((lambda : print('b')))
unittest.addModuleCleanup((lambda : print('c')))
Résultat d'exécution
c
b
a
createConnection
E
Elle est enregistrée dans l'ordre a-> b-> c et sortie dans l'ordre c-> b-> a.
La fonction enregistrée est exécutée par la fonction doModuleCleanups. Le pointu a peut-être remarqué que lors de l'exécution dans LIFO, il apparaît dans la liste. Donc, si vous exécutez d'abord la fonction doModuleCleanups, rien ne sera exécuté car la liste des fonctions à exécuter est un shell.
Ajoutez le code suivant à la partie commentaire TODO du code ci-dessus.
unittest.addModuleCleanup((lambda : print('a')))
unittest.addModuleCleanup((lambda : print('b')))
unittest.addModuleCleanup((lambda : print('c')))
unittest.case.doModuleCleanups()
print()
Résultat de sortie
c
b
a
createConnection
E
Une exception s'est produite après qu'une ligne a été libérée par print () ''.
La fonction réenregistrée ne sera pas exécutée.
De plus, bien qu'il s'agisse d'une fonction doModuleCleanups, `case``` est omis dans la documentation officielle, vous devriez donc l'appeler` `ʻunittest.case.doModuleCleanups ()`. (Mauvaise description?)
Il n'aurait dû y avoir aucun article individuel (au 4 mai 2020), alors j'en ai créé un. N'a-t-il pas l'impression que cette fonctionnalité n'est pas disponible dans pytest? J'espère que tu trouves cela utile.
Si vous êtes gentil, veuillez répondre aux questions suivantes ← ・ ・ ・
Lors de l'enregistrement de plusieurs fonctions, j'ai écrit à l'origine comme suit, Puisque l'id est le même, je produis lambda pour chaque abc comme l'article. Comment puis-je créer des instances individuelles ...? (Est-ce du Jamuripo en une seule ligne?)
#Avec cela, tous les 4 sont émis ...
for i in range(5):
unittest.addModuleCleanup((lambda : print(i)))
URL
--Veuillez vous rendre au responsable pour une explication détaillée → https://docs.python.org/ja/3/library/unittest.html
Recommended Posts