tidy up the tests

Signed-off-by: Michael Donaghy <md401@srcf.ucam.org>