From f42bfecb97f32ac18c0424a862a0be7101878cda Mon Sep 17 00:00:00 2001
From: Kengo Seki <sekikn@apache.org>
Date: Thu, 28 Jun 2018 13:37:53 -0400
Subject: [PATCH] [AIRFLOW-2687] Move unittest2 to devel in setup.py

This PR moves the package unittest2 from
devel_ci to devel in setup.py, since
the latter is the minimal required packages
for development and unittest2 is considered
as one of them.
---
 setup.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 798318c7d99aa..4a63010b2da26 100644
--- a/setup.py
+++ b/setup.py
@@ -218,6 +218,10 @@ def write_version(filename=os.path.join(*['airflow',
     'rednose',
     'requests_mock'
 ]
+
+if not PY3:
+    devel += ['unittest2']
+
 devel_minreq = devel + kubernetes + mysql + doc + password + s3 + cgroups
 devel_hadoop = devel_minreq + hive + hdfs + webhdfs + kerberos
 devel_all = (sendgrid + devel + all_dbs + doc + samba + s3 + slack + crypto + oracle +
@@ -231,7 +235,7 @@ def write_version(filename=os.path.join(*['airflow',
     devel_ci = [package for package in devel_all if package not in
                 ['snakebite>=2.7.8', 'snakebite[kerberos]>=2.7.8']]
 else:
-    devel_ci = devel_all + ['unittest2']
+    devel_ci = devel_all
 
 
 def do_setup():