rm NoneType

narwhals-dev · Sep 13, 2024 · eb1468e · eb1468e
1 parent cd2ed40
commit eb1468e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/dtypes_test.py b/tests/dtypes_test.py
@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 from datetime import timezone
-from types import NoneType
 from typing import Literal
 
 import pytest
@@ -17,7 +16,7 @@ def test_datetime_valid(
     dtype = nw.Datetime(time_unit=time_unit, time_zone=time_zone)
 
     assert dtype.time_unit == time_unit
-    assert isinstance(dtype.time_zone, (str, NoneType))
+    assert isinstance(dtype.time_zone, str) or dtype.time_zone is None
 
 
 @pytest.mark.parametrize("time_unit", ["abc", "s"])