Update tests.py

Custom user model