Add helper methods to tests.py