Support python 3

Signed-off-by: Daniel Nephin <dnephin@gmail.com>

1 files changed, 4 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 2f6dad7a..b7fd4403 100644
--- a/setup.py
+++ b/setup.py
@@ -48,8 +48,11 @@ tests_require = [
 ]
 
 
-if sys.version_info < (2, 7):
+if sys.version_info < (2, 6):
     tests_require.append('unittest2')
+if sys.version_info[:1] < (3,):
+    tests_require.append('pyinstaller')
+    tests_require.append('mock >= 1.0.1')
 
 
 setup(