From ca90b6dee68c90c1e824d1ece9c565c374058910 Mon Sep 17 00:00:00 2001 From: Khaleel Al-Adhami Date: Mon, 4 Nov 2024 11:04:06 -0800 Subject: [PATCH] older version of python doesn't have assert type --- tests/units/vars/test_object.py | 25 ++++++++++++++----------- 1 file changed, 14 insertions(+), 11 deletions(-) diff --git a/tests/units/vars/test_object.py b/tests/units/vars/test_object.py index 019c8ac4b..60dffd7e3 100644 --- a/tests/units/vars/test_object.py +++ b/tests/units/vars/test_object.py @@ -1,4 +1,4 @@ -from typing import assert_type +import sys import pytest @@ -94,16 +94,19 @@ def test_state_to_operation(type_: GenericType) -> None: def test_typing() -> None: - # Bare - var = ObjectState.bare.to(ObjectVar) - _ = assert_type(var, ObjectVar[Bare]) + if sys.version_info >= (3, 11): + from typing import assert_type - var = ObjectState.base.to(ObjectVar, Base) - _ = assert_type(var, ObjectVar[Base]) + # Bare + var = ObjectState.bare.to(ObjectVar) + _ = assert_type(var, ObjectVar[Bare]) - # Base - var = ObjectState.base.to(ObjectVar) - _ = assert_type(var, ObjectVar[Base]) + var = ObjectState.base.to(ObjectVar, Base) + _ = assert_type(var, ObjectVar[Base]) - var = ObjectState.base.to(LiteralObjectVar, Base) - _ = assert_type(var, ObjectVar[Base]) + # Base + var = ObjectState.base.to(ObjectVar) + _ = assert_type(var, ObjectVar[Base]) + + var = ObjectState.base.to(LiteralObjectVar, Base) + _ = assert_type(var, ObjectVar[Base])