diff --git a/tests/test_telemetry.py b/tests/test_telemetry.py index 98e5d558a..4ba9ec8c7 100644 --- a/tests/test_telemetry.py +++ b/tests/test_telemetry.py @@ -1,10 +1,8 @@ +from packaging.version import parse as parse_python_version + from reflex.utils import telemetry -def versiontuple(v): - return tuple(map(int, (v.split(".")))) - - def test_telemetry(): """Test that telemetry is sent correctly.""" # Check that the user OS is one of the supported operating systems. @@ -24,7 +22,7 @@ def test_telemetry(): # Check that the Python version is greater than 3.7. python_version = telemetry.get_python_version() assert python_version is not None - assert versiontuple(python_version) >= versiontuple("3.7") + assert parse_python_version(python_version) >= parse_python_version("3.7") def test_disable():