Add test for various ClassVars

python-attrs · Feb 19, 2021 · 0d5ecdc · 0d5ecdc
1 parent 0c16aec
commit 0d5ecdc
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/test_annotations.py b/tests/test_annotations.py
@@ -12,6 +12,7 @@
 
 import attr
 
+from attr._make import _is_class_var
 from attr.exceptions import UnannotatedAttributeError
 
 
@@ -604,3 +605,19 @@ class C:
 
         with pytest.raises(NameError):
             typing.get_type_hints(C.__init__)
+
+
+@pytest.mark.parametrize(
+    "annot",
+    [
+        typing.ClassVar,
+        "typing.ClassVar",
+        "'typing.ClassVar[dict]'",
+        "t.ClassVar[int]",
+    ],
+)
+def test_is_class_var(annot):
+    """
+    ClassVars are detected, even if they're a string or quoted.
+    """
+    assert _is_class_var(annot)