def test_should_show_new_version_notice_prerelease(self):
with mock.patch(
"streamlit.version._get_latest_streamlit_version"
) as get_latest, mock.patch(
"streamlit.version._get_installed_streamlit_version"
) as get_installed:
version.CHECK_PYPI_PROBABILITY = 1
get_installed.side_effect = [PkgVersion("1.2.3")]
get_latest.side_effect = [PkgVersion("1.0.0")]
self.assertFalse(should_show_new_version_notice())
get_installed.assert_called_once()
get_latest.assert_called_once()
def test_get_latest_streamlit_version(self):
with requests_mock.mock() as m:
m.get(PYPI_STREAMLIT_URL, json={"info": {"version": "1.2.3"}})
self.assertEqual(PkgVersion("1.2.3"),
_get_latest_streamlit_version())