Refine test helper function
Refine test helper function

This commit contains refinements of error messages, including by
adding color to prompts (using the colorama package, which should take
care of compatibility across platforms).

StanczakDominik/PlasmaPy