unittest2 is only a test requirement

Author: asottile

Diff for: .travis.yml

@@ -9,8 +9,9 @@ env:
   - DB=mysql57
   - DB=mysql56
 install:
-  - "pip install ."
-  - "pip install nose"
+  # Needs a newer version of pip to do the pip installation line
+  - pip install pip --upgrade
+  - pip install .[test] nose
 cache: apt
 before_script:
   - env | grep DB

Diff for: setup.py

@@ -8,11 +8,11 @@
 
 import sys
 
-install_requires = ['pymysql']
+tests_require = []
 
-# add unittest2 to install_requires for python < 2.7
+# add unittest2 to tests_require for python < 2.7
 if sys.version_info < (2, 7):
-    install_requires.append("unittest2")
+    tests_require.append("unittest2")
 
 
 class TestCommand(Command):
@@ -49,5 +49,6 @@ def run(self):
               "pymysqlreplication.constants",
               "pymysqlreplication.tests"],
     cmdclass={"test": TestCommand},
-    install_requires=install_requires,
+    extras_require={'test': tests_require},
+    install_requires=['pymysql'],
 )