Move tests to test.py.