Tests + cleanups for assert_equals

This version even supports coloured output! Should make debugging
easier :)