diff --git a/naive-nlu/tree_nlu/test.py b/naive-nlu/tree_nlu/test.py index 021e2fc..e6c0102 100644 --- a/naive-nlu/tree_nlu/test.py +++ b/naive-nlu/tree_nlu/test.py @@ -13,11 +13,11 @@ def main(): for test_name, test_module in tests: try: test_module.main() - print(" ✓ {}".format(test_name)) + print(" \x1b[1;32m✓\x1b[0m {}".format(test_name)) except AssertionError as ae: - print(" ✗ {}: {}".format(test_name, ae.args[0])) + print(" \x1b[1;31m✗\x1b[0m {}: {}".format(test_name, ae.args[0])) except Exception as e: - print(" ! {} {}".format(test_name, e)) + print(" \x1b[1;7;31m!\x1b[0m {} {}".format(test_name, e)) raise if __name__ == '__main__':