diff --git a/zephyr/tests.py b/zephyr/tests.py index 99ab98a8b7..d48c5b33e0 100644 --- a/zephyr/tests.py +++ b/zephyr/tests.py @@ -21,13 +21,19 @@ import re import sys import random +def bail(msg): + print '\nERROR: %s\n' % (msg,) + sys.exit(1) + try: settings.TEST_SUITE except: - print - print "ERROR: Test suite only runs correctly with --settings=humbug.test_settings" - print - sys.exit(1) + bail('Test suite only runs correctly with --settings=humbug.test_settings') + +try: + import pygments +except ImportError: + bail('The Pygments library is required to run the backend test suite.') def find_key_by_email(address): from django.core.mail import outbox