Omit relative imports from checking

Author: arkq

src/flake8_requirements/checker.py

@@ -36,6 +36,9 @@ def visit_Import(self, node):
         self.imports.append((node, node.names[0].name))
 
     def visit_ImportFrom(self, node):
+        if node.level != 0:
+            # Omit relative imports (local modules).
+            return
         self.imports.append((node, node.module))
 
 

test/test_checker.py

@@ -60,3 +60,17 @@ def test_3rd_party_hyphen(self):
     def test_3rd_party_multi_module(self):
         errors = check("import pkg_resources")
         self.assertEqual(len(errors), 0)
+
+    def test_non_top_level_import(self):
+        errors = check("def function():\n import cat")
+        self.assertEqual(len(errors), 1)
+        self.assertEqual(
+            errors[0][2],
+            "I900 'cat' not listed as a requirement",
+        )
+
+    def test_relative_import(self):
+        errors = check("from . import local")
+        self.assertEqual(len(errors), 0)
+        errors = check("from ..local import local")
+        self.assertEqual(len(errors), 0)