diff options
Diffstat (limited to 'tests/pyreverse/test_writer.py')
-rw-r--r-- | tests/pyreverse/test_writer.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/tests/pyreverse/test_writer.py b/tests/pyreverse/test_writer.py index 2897ca054..37a4b4f19 100644 --- a/tests/pyreverse/test_writer.py +++ b/tests/pyreverse/test_writer.py @@ -35,6 +35,7 @@ _DEFAULTS = { "show_stdlib": False, "only_classnames": False, "output_directory": "", + "no_standalone": False, } TEST_DATA_DIR = os.path.join(os.path.dirname(__file__), "..", "data") @@ -45,6 +46,7 @@ PUML_FILES = ["packages_No_Name.puml", "classes_No_Name.puml"] COLORIZED_PUML_FILES = ["packages_colorized.puml", "classes_colorized.puml"] MMD_FILES = ["packages_No_Name.mmd", "classes_No_Name.mmd"] HTML_FILES = ["packages_No_Name.html", "classes_No_Name.html"] +NO_STANDALONE_FILES = ["classes_no_standalone.dot", "packages_no_standalone.dot"] class Config: @@ -88,6 +90,15 @@ def setup_colorized_dot( @pytest.fixture() +def setup_no_standalone_dot( + no_standalone_dot_config: PyreverseConfig, get_project: GetProjectCallable +) -> Iterator[None]: + writer = DiagramWriter(no_standalone_dot_config) + project = get_project(TEST_DATA_DIR, name="no_standalone") + yield from _setup(project, no_standalone_dot_config, writer) + + +@pytest.fixture() def setup_puml( puml_config: PyreverseConfig, get_project: GetProjectCallable ) -> Iterator[None]: @@ -138,6 +149,7 @@ def _setup( for fname in ( DOT_FILES + COLORIZED_DOT_FILES + + NO_STANDALONE_FILES + PUML_FILES + COLORIZED_PUML_FILES + MMD_FILES @@ -161,6 +173,12 @@ def test_colorized_dot_files(generated_file: str) -> None: _assert_files_are_equal(generated_file) +@pytest.mark.usefixtures("setup_no_standalone_dot") +@pytest.mark.parametrize("generated_file", NO_STANDALONE_FILES) +def test_no_standalone_dot_files(generated_file: str) -> None: + _assert_files_are_equal(generated_file) + + @pytest.mark.usefixtures("setup_puml") @pytest.mark.parametrize("generated_file", PUML_FILES) def test_puml_files(generated_file: str) -> None: |