colorful nosetests wrapper

cnose

#!/usr/bin/env bash

set -euo pipefail

simple() {
	sed -E \
		-e "1 s/\./$(printf '\033[32m.\033[0m')/g" \
		-e "1 s/S/$(printf '\033[33mS\033[0m')/g" \
		-e "1 s/F/$(printf '\033[31mF\033[0m')/g" \
		-e "s/^(FAILED.*)$/$(printf '\033[31m\\1\033[0m')/" \
		-e "s/^(OK.*)$/$(printf '\033[32m\\1\033[0m')/"
}

verbose() {
	sed -E \
		-e "s/^(.*\.\.\. ok)$/$(printf '\033[32m\\1\033[0m')/" \
		-e "s/^(.*\.\.\. SKIP: .*)$/$(printf '\033[33m\\1\033[0m')/" \
		-e "s/^(.*\.\.\. FAIL)$/$(printf '\033[31m\\1\033[0m')/" \
		-e "s/^(FAILED.*)$/$(printf '\033[31m\\1\033[0m')/" \
		-e "s/^(OK.*)$/$(printf '\033[32m\\1\033[0m')/"
}

VERBOSE=
for arg in $@; do
	if [[ "$arg" = "-v" ]]; then
		VERBOSE=1
	fi
done

if [[ "$VERBOSE" ]]; then
	nosetests "$@" 2>&1| verbose
else
	nosetests "$@" 2>&1| simple
fi