Create tests.py

Create the tests here.