Update testing framework for Travis