Fix tests

Removed test data from separate file and put it in dict