From 034b6779d5920ccdaf401ba159e879c899b9694d Mon Sep 17 00:00:00 2001 From: Khaleel Al-Adhami Date: Mon, 4 Nov 2024 13:31:13 -0800 Subject: [PATCH] get it from typing_extension --- tests/units/vars/test_object.py | 18 +++++++----------- 1 file changed, 7 insertions(+), 11 deletions(-) diff --git a/tests/units/vars/test_object.py b/tests/units/vars/test_object.py index 1b156229f..efcb21166 100644 --- a/tests/units/vars/test_object.py +++ b/tests/units/vars/test_object.py @@ -1,6 +1,5 @@ -import sys - import pytest +from typing_extensions import assert_type import reflex as rx from reflex.utils.types import GenericType @@ -94,13 +93,10 @@ def test_state_to_operation(type_: GenericType) -> None: def test_typing() -> None: - if sys.version_info >= (3, 11): - from typing import assert_type + # Bare + var = ObjectState.bare.to(ObjectVar) + _ = assert_type(var, ObjectVar[Bare]) - # Bare - var = ObjectState.bare.to(ObjectVar) - _ = assert_type(var, ObjectVar[Bare]) - - # Base - var = ObjectState.base - _ = assert_type(var, ObjectVar[Base]) + # Base + var = ObjectState.base + _ = assert_type(var, ObjectVar[Base])