Add more tests

Signed-off-by: David Fisher-Moreau <deiform87@gmail.com>