Update tests

[notionparallax]

Either use a 'real' test framework, or update them to look like the exam tests.
A real framework would probably be best, but the exams are nicely verbose.