write some tests

These tests were messed up!

I'm a little rusty, but hopefully this is better than before :+1: