Add trait for getting all unit tests of a module.

This will also make it possible to actually call the unit test functions.