Create tests.py