Collect tests requirements

mwgamble · Nov 29, 2017 · f3aab51 · f3aab51
1 parent e7387a0
commit f3aab51
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/flake8_requirements/checker.py b/src/flake8_requirements/checker.py
@@ -166,7 +166,8 @@ def setup(**kw):
         # Restore import search path.
         sys.path.pop(0)
 
-    def get_requirements(self, install=True, extras=True, setup=False):
+    def get_requirements(
+            self, install=True, extras=True, setup=False, tests=False):
         """Get package requirements."""
         requires = []
         if install:
@@ -180,6 +181,10 @@ def get_requirements(self, install=True, extras=True, setup=False):
             requires.extend(parse_requirements(
                 self.keywords.get('setup_requires', ()),
             ))
+        if tests:
+            requires.extend(parse_requirements(
+                self.keywords.get('tests_require', ()),
+            ))
         return requires
 
     def visit_Call(self, node):
@@ -309,6 +314,7 @@ def modcmp(lib=(), test=()):
 
         requirements = self.setup.get_requirements(
             setup=self.processing_setup_py,
+            tests=True,
         )
 
         # Get 3rd party module names based on requirements.