[AIRFLOW-911] Add coloring and timing to tests

Closes apache#2106 from bolkedebruin/profile_tests
dgies · Feb 25, 2017 · 4f52db3 · 4f52db3
1 parent bc47200
commit 4f52db3
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/run_unit_tests.sh b/run_unit_tests.sh
@@ -46,6 +46,8 @@ if [ "${TRAVIS}" ]; then
     --cover-package=airflow \
     --cover-html-dir=airflow/www/static/coverage \
     --with-ignore-docstrings \
+    --rednose \
+    --with-timer \
     -v \
     --logging-level=DEBUG "
     fi
@@ -68,6 +70,8 @@ else
     --cover-package=airflow \
     --cover-html-dir=airflow/www/static/coverage \
     --with-ignore-docstrings \
+    --rednose \
+    --with-timer \
     -s \
     -v \
     --logging-level=DEBUG "

diff --git a/scripts/ci/requirements.txt b/scripts/ci/requirements.txt
@@ -39,6 +39,7 @@ nose
 nose-exclude
 nose-ignore-docstring==0.2
 nose-parameterized
+nose-timer
 pandas
 psutil>=4.2.0, <5.0.0
 psycopg2
@@ -51,6 +52,7 @@ PySmbClient
 python-daemon
 python-dateutil
 redis
+rednose
 requests
 requests-kerberos
 setproctitle

diff --git a/setup.py b/setup.py
@@ -180,6 +180,8 @@ def write_version(filename=os.path.join(*['airflow',
     'nose',
     'nose-ignore-docstring==0.2',
     'nose-parameterized',
+    'nose-timer',
+    'rednose'
 ]
 devel_minreq = devel + mysql + doc + password + s3 + cgroups
 devel_hadoop = devel_minreq + hive + hdfs + webhdfs + kerberos