put this method into tests.py