From 1c24da23aea44e9f0be44c69d8650f63f0e7a841 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Barrois?= <raphael.barrois@polytechnique.org>
Date: Wed, 14 Aug 2013 17:56:08 +0200
Subject: More fixes to tests

---
 setup.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'setup.py')

diff --git a/setup.py b/setup.py
index f9c2844..1b49342 100755
--- a/setup.py
+++ b/setup.py
@@ -3,6 +3,7 @@
 
 import os
 import re
+import sys
 
 from setuptools import setup
 
@@ -21,6 +22,11 @@ def get_version(package_name):
     return '0.1.0'
 
 
+if sys.version_info[0:2] < (2, 7):  # pragma: no cover
+    test_loader = 'unittest2:TestLoader'
+else:
+    test_loader = 'unittest:TestLoader'
+
 
 PACKAGE = 'factory'
 
@@ -40,6 +46,9 @@ setup(
     setup_requires=[
         'setuptools>=0.8',
     ],
+    tests_require=[
+        'mock',
+    ],
     classifiers=[
         'Development Status :: 5 - Production/Stable',
         'Intended Audience :: Developers',
@@ -58,4 +67,5 @@ setup(
         'Topic :: Software Development :: Libraries :: Python Modules'
     ],
     test_suite='tests',
+    test_loader=test_loader,
 )
-- 
cgit v1.2.3