tests/test_page_2.py

from streamlit.testing.v1 import AppTest
from tests.pwd_protection import pwd_protection, pwd_protection_pwd_incorrect

def test_pwd_protection(page_2: AppTest):
    pwd_protection(page_2)

def test_pwd_protection_pwd_incorrect(page_2: AppTest):
    pwd_protection_pwd_incorrect(page_2)