import streamlit as st

from common.streamlit_utilities import reset_form


def test_reset_form()-> None:
    """
    Test Streamlit functionality to clear all session objects
    """

    st.session_state.fan_city_value_key = "London"
    st.session_state.first_name_column_key = True

    reset_form()

    assert {} == {k: v for k, v in st.session_state.items() if v not in [None, False]}


def test_reset_form2()-> None:
    """
    Test Streamlit functionality to clear all session objects
    """

    st.session_state.fan_city_value_key = "London"
    st.session_state.first_name_column_key = True

    reset_form()

    assert {} == {k: v for k, v in st.session_state.items() if v not in [None, False]}