adding some fanciness from unittest