Add custom User model support to tests.py