blob: a2da6e27ddd30f8ec7d1c3a45affda62d32a726b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
# Copyright (c) 2021 Takahide Nojima <nozzy123nozzy@gmail.com>
#
# Licensed under the GPL: https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
# For details: https://github.com/PyCQA/pylint/blob/main/LICENSE
"""Unit test for the diagrams modules"""
# pylint: disable=redefined-outer-name
from typing import Callable
from pylint.pyreverse.diadefslib import DefaultDiadefGenerator, DiadefsHandler
from pylint.pyreverse.inspector import Linker
from pylint.testutils.pyreverse import PyreverseConfig
def test_property_handling(
default_config: PyreverseConfig, get_project: Callable
) -> None:
project = get_project("data.property_pattern")
class_diagram = DefaultDiadefGenerator(
Linker(project), DiadefsHandler(default_config)
).visit(project)[0]
obj = class_diagram.classe("PropertyPatterns")
assert len(class_diagram.get_methods(obj.node)) == 0
assert class_diagram.get_attrs(obj.node) == ["prop1", "prop2"]
|