Tests added

For me, and this is a personal opinion, it's not a good idea to be tight to data on the tests