TST: Adds some tests