A tiny little unit test framework to make life easier in ML