Move the test case to the tests.py file