diff options
| author | Marc Mueller <30130371+cdce8p@users.noreply.github.com> | 2021-02-28 18:39:50 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-02-28 18:39:50 +0100 |
| commit | 5f2cfb854c6ab8ae075f17e88d47611954d91ff8 (patch) | |
| tree | 60f27670361a25f35ea2b157b004a9975b6b12c9 /tests | |
| parent | 0f8e2337556832889b400b8f570390eca0a51817 (diff) | |
| download | astroid-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.py | 12 |
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", |
