moved tests to test.py