Faciliter la lecture de parametrize dans pytest

Lorsque l'on a besoin de tester une fonction il est intéressant de le faire sur une multitude de valeurs gérant les cas nominaux, aux limites et d'erreurs.
Pour cela nous allons garder la même fonction de test

Copier le code
1
2
3
4
5
def test_user__check_attributs(id,name,birthday,country):
 assert type(id) is int 
 assert type(name) is str 
 assert type(datetime.strptime(birthday, "%d/%m/%Y")) is datetime 
 assert type(country) is str

Mais le jeu de données sera différent. Partons du principe que l'on veut avoir cet objet

Copier le code
1
2
3
4
5
6
@dataclass
class User:
    id: int
    name: str 
    birthday: datetime
    country: str

Et avec la fonction pour tester son contenu

Copier le code
1
2
3
4
5
6
@pytest.mark.parametrize("id,name,birthday,country", [(28, "Katnis", "14/10/1987", "France")])
def test_user__check_attributs(id,name,birthday,country):
 assert type(id) is int 
 assert type(name) is str 
 assert type(datetime.strptime(birthday, "%d/%m/%Y")) is datetime 
 assert type(country) is str

La fonction

parametrize

Elle prend une liste (plus précisément un

tuple)

Copier le code
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
@pytest.mark.parametrize("id,name,birthday,country", [
      (28, "Katnis", "14/10/1987", "France"),
      (12, "Emy", "12/12/2007", "France"),
      (46, "Paul", "10/02/1987", "France"),
      (100, "Fred", "09/12/1995", "France"),
      (3, "Henri", "12/07/1957", "France"),
])
def test_user__check_attributs(id,name,birthday,country):
 assert type(id) is int 
 assert type(name) is str 
 assert type(datetime.strptime(birthday, "%d/%m/%Y")) is datetime 
 assert type(country) is str

Vous imaginez que notre objet

User

parametrize

Copier le code
1
2
3
4
5
6
7
USERS = [
    User(28, "Katnis", "14/10/1987", "France"),
    User(12, "Emy", "12/12/2007", "France"),
    User(46, "Paul", "10/02/1987", "France"),
    User(100, "Fred", "09/12/1995", "France"),
    User(3, "Henri", "12/07/1957", "France"),
]

On souhaite injecter ces données dans notre fonction parametrize de cette manière

Copier le code
1
2
3
4
@pytest.mark.parametrize("id,name,birthday,country", [
       pytest.param(*current_user.__dict__.values())
       for current_user in USERS
])

• On crée une compréhension de liste qui itère sur notre jeu de données

  USERS

• On utilise la fonction

  pytest.param

  qui prend l'objet

  current_user

  en cours d'itération pour le convertir en dictionnaire (

  __dict__

  ) et lui extraire ces valeurs via

  values()

  (par défaut le parcours d'un dictionnaire retourne les clés). On fait du unpacking pour envoyer ces valeurs aux paramètres respectifs "id,name,birthday,country"

Ce qui donne au final

Copier le code
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
USERS = [
 User(28, "Katnis", "14/10/1987", "France"),
 User(12, "Emy", "12/12/2007", "France"),
 User(46, "Paul", "10/02/1987", "France"),
 User(100, "Fred", "09/12/1995", "France"),
 User(3, "Henri", "12/07/1957", "France"),
]

@pytest.mark.parametrize("id,name,birthday,country", [
 pytest.param(*current_user.__dict__.values())
 for current_user in USERS
])
def test_user__check_attributs(id,name,birthday,country):
 assert type(id) is int
 assert type(name) is str
 assert type(datetime.strptime(birthday, "%d/%m/%Y")) is datetime
 assert type(country) is str

Ainsi la fonction de test

test_user__check_attributs