diff --git a/requirements-test.txt b/requirements-test.txt
new file mode 100644
index 0000000..f3c7e8e
--- /dev/null
+++ b/requirements-test.txt
@@ -0,0 +1 @@
+nose
diff --git a/setup.py b/setup.py
index d0acfe0..a6e77ef 100755
--- a/setup.py
+++ b/setup.py
@@ -1,30 +1,42 @@
-from setuptools import setup
+#!/usr/bin/env python3
 
+import os
+
+from setuptools import setup, find_packages
+
+
+def parse_requirements(name=None):
+    if name:
+        reqf = 'requirements-%s.txt' % name
+    else:
+        reqf = 'requirements.txt'
 
-def parse_requirements():
     requirements = []
-    for reqf in ('requirements.txt', 'requirements-swh.txt'):
-        with open(reqf) as f:
-            for line in f.readlines():
-                line = line.strip()
-                if not line or line.startswith('#'):
-                    continue
-                requirements.append(line)
+    if not os.path.exists(reqf):
+        return requirements
+
+    with open(reqf) as f:
+        for line in f.readlines():
+            line = line.strip()
+            if not line or line.startswith('#'):
+                continue
+            requirements.append(line)
     return requirements
 
 
 # Edit this part to match your module
 # full sample: https://forge.softwareheritage.org/diffusion/DCORE/browse/master/setup.py
 setup(
     name='swh.<module-name>',  # example: swh.loader.pypi
     description='Software Heritage <Module\'s intent>',
     author='Software Heritage developers',
     author_email='swh-devel@inria.fr',
     url='https://forge.softwareheritage.org/diffusion/<module-git-code>',
-    packages=[],  # packages's modules
+    packages=find_packages(),  # packages's modules
     scripts=[],   # scripts to package
-    install_requires=parse_requirements(),
+    install_requires=parse_requirements() + parse_requirements('swh'),
+    test_requires=parse_requirements('test'),
     setup_requires=['vcversioner'],
     vcversioner={},
     include_package_data=True,
 )