Remove print statement from tests.py