Add test.py for travis CI