changed tests.py to import the test