Tests!

Fully tested!

Includes:

• Code coverage (99%)
• Full tests that run multiple models in different configs
• Full tests that test specific functionality in trainer.