Tests: Set up tests to work with Travis CI