diff --git a/tests/auth_tests/models/custom_user.py b/tests/auth_tests/models/custom_user.py index 0519b09b75a..62dbf9c8be0 100644 --- a/tests/auth_tests/models/custom_user.py +++ b/tests/auth_tests/models/custom_user.py @@ -3,6 +3,7 @@ from django.contrib.auth.models import ( PermissionsMixin, UserManager, ) from django.db import models +from django.utils.encoding import python_2_unicode_compatible # The custom User uses email as the unique identifier, and requires @@ -32,6 +33,7 @@ class CustomUserManager(BaseUserManager): return u +@python_2_unicode_compatible class CustomUser(AbstractBaseUser): email = models.EmailField(verbose_name='email address', max_length=255, unique=True) is_active = models.BooleanField(default=True) @@ -49,7 +51,7 @@ class CustomUser(AbstractBaseUser): def get_short_name(self): return self.email - def __unicode__(self): + def __str__(self): return self.email # Maybe required?