Stopped installing tests with package

1 files changed, 83 insertions, 0 deletions

diff --git a/tests/test_import_graph.py b/tests/test_import_graph.py
new file mode 100644
index 000000000..d7b009693
--- /dev/null
+++ b/tests/test_import_graph.py
@@ -0,0 +1,83 @@
+# Copyright (c) 2006-2008, 2010, 2013 LOGILAB S.A. (Paris, FRANCE) <contact@logilab.fr>
+# Copyright (c) 2012 FELD Boris <lothiraldan@gmail.com>
+# Copyright (c) 2014-2017 Claudiu Popa <pcmanticore@gmail.com>
+# Copyright (c) 2014 Google, Inc.
+# Copyright (c) 2015 Ionel Cristian Maries <contact@ionelmc.ro>
+# Copyright (c) 2016 Derek Gustafson <degustaf@gmail.com>
+# Copyright (c) 2018 Reverb C <reverbc@users.noreply.github.com>
+
+# Licensed under the GPL: https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
+# For details: https://github.com/PyCQA/pylint/blob/master/COPYING
+
+import os
+from os.path import exists
+
+import pytest
+
+import pylint.testutils as testutils
+from pylint.checkers import imports, initialize
+from pylint.lint import PyLinter
+
+
+@pytest.fixture
+def dest():
+    dest = "dependencies_graph.dot"
+    yield dest
+    os.remove(dest)
+
+
+def test_dependencies_graph(dest):
+    imports._dependencies_graph(dest, {"labas": ["hoho", "yep"], "hoho": ["yep"]})
+    with open(dest) as stream:
+        assert (
+            stream.read().strip()
+            == """
+digraph "dependencies_graph" {
+rankdir=LR
+charset="utf-8"
+URL="." node[shape="box"]
+"hoho" [];
+"yep" [];
+"labas" [];
+"yep" -> "hoho" [];
+"hoho" -> "labas" [];
+"yep" -> "labas" [];
+}
+""".strip()
+        )
+
+
+@pytest.fixture
+def linter():
+    l = PyLinter(reporter=testutils.TestReporter())
+    initialize(l)
+    return l
+
+
+@pytest.fixture
+def remove_files():
+    yield
+    for fname in ("import.dot", "ext_import.dot", "int_import.dot"):
+        try:
+            os.remove(fname)
+        except:
+            pass
+
+
+@pytest.mark.usefixtures("remove_files")
+def test_checker_dep_graphs(linter):
+    l = linter
+    l.global_set_option("persistent", False)
+    l.global_set_option("reports", True)
+    l.global_set_option("enable", "imports")
+    l.global_set_option("import-graph", "import.dot")
+    l.global_set_option("ext-import-graph", "ext_import.dot")
+    l.global_set_option("int-import-graph", "int_import.dot")
+    l.global_set_option("int-import-graph", "int_import.dot")
+    # ignore this file causing spurious MemoryError w/ some python version (>=2.3?)
+    l.global_set_option("ignore", ("func_unknown_encoding.py",))
+    l.check("input")
+    l.generate_reports()
+    assert exists("import.dot")
+    assert exists("ext_import.dot")
+    assert exists("int_import.dot")