1- from collections import Generator
1+ from typing import Generator
22
33import pytest
44from sqlalchemy .orm import Session
1010@pytest .fixture
1111def user (session : Session ) -> Generator [User , None , None ]:
1212 mock_user = create_model (
13- session , User ,
14- username = 'test_username' ,
15- password = 'test_password' ,
16- email = 'test.email@gmail.com' ,
13+ session ,
14+ User ,
15+ username = "test_username" ,
16+ password = "test_password" ,
17+ email = "test.email@gmail.com" ,
1718 language_id = 1 ,
1819 )
1920 yield mock_user
@@ -23,10 +24,11 @@ def user(session: Session) -> Generator[User, None, None]:
2324@pytest .fixture
2425def sender (session : Session ) -> Generator [User , None , None ]:
2526 mock_user = create_model (
26- session , User ,
27- username = 'sender_username' ,
28- password = 'sender_password' ,
29- email = 'sender.email@gmail.com' ,
27+ session ,
28+ User ,
29+ username = "sender_username" ,
30+ password = "sender_password" ,
31+ email = "sender.email@gmail.com" ,
3032 language_id = 1 ,
3133 )
3234 yield mock_user
0 commit comments