[TASK] Update tests for usage with Travis CI