Improve tests

Signed-off-by: Daniel Francis <francis@sugarlabs.org>