summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorMarc Mueller <30130371+cdce8p@users.noreply.github.com>2021-02-28 18:39:50 +0100
committerGitHub <noreply@github.com>2021-02-28 18:39:50 +0100
commit5f2cfb854c6ab8ae075f17e88d47611954d91ff8 (patch)
tree60f27670361a25f35ea2b157b004a9975b6b12c9 /tests
parent0f8e2337556832889b400b8f570390eca0a51817 (diff)
downloadastroid-git-5f2cfb854c6ab8ae075f17e88d47611954d91ff8.tar.gz
Don't transform dataclass ClassVars (#914)
Co-authored-by: Pierre Sassoulas <pierre.sassoulas@gmail.com>
Diffstat (limited to 'tests')
-rw-r--r--tests/unittest_brain.py12
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/unittest_brain.py b/tests/unittest_brain.py
index 40ee7dc3..b0a97c55 100644
--- a/tests/unittest_brain.py
+++ b/tests/unittest_brain.py
@@ -2108,6 +2108,8 @@ def test_dataclasses():
code = """
import dataclasses
from dataclasses import dataclass
+ import typing
+ from typing import ClassVar
@dataclass
class InventoryItem:
@@ -2117,6 +2119,8 @@ def test_dataclasses():
@dataclasses.dataclass
class Other:
name: str
+ CONST_1: ClassVar[int] = 42
+ CONST_2: typing.ClassVar[int] = 42
"""
module = astroid.parse(code)
@@ -2135,6 +2139,14 @@ def test_dataclasses():
assert len(name) == 1
assert isinstance(name[0], astroid.Unknown)
+ const_1 = second.getattr("CONST_1")
+ assert len(const_1) == 1
+ assert isinstance(const_1[0], astroid.AssignName)
+
+ const_2 = second.getattr("CONST_2")
+ assert len(const_2) == 1
+ assert isinstance(const_2[0], astroid.AssignName)
+
@pytest.mark.parametrize(
"code,expected_class,expected_value",