From 9d62b992fb53ecd880718622dd2c6ba5abf1c413 Mon Sep 17 00:00:00 2001 From: Arnav Choudhury Date: Sat, 30 Jan 2021 17:18:28 +0530 Subject: [PATCH] Reverted to using absolute imports to fetch the User model since mypy will fail as type will be determined at run-time --- .../{{cookiecutter.project_slug}}/users/tests/test_models.py | 4 ++-- .../{{cookiecutter.project_slug}}/users/tests/test_urls.py | 4 ++-- .../{{cookiecutter.project_slug}}/users/tests/test_views.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_models.py b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_models.py index e7180db16..3194be1fd 100644 --- a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_models.py +++ b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_models.py @@ -1,8 +1,8 @@ import pytest -from django.contrib.auth import get_user_model + +from {{ cookiecutter.project_slug }}.users.models import User pytestmark = pytest.mark.django_db -User = get_user_model() def test_user_get_absolute_url(user: User): diff --git a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_urls.py b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_urls.py index f5cc874a4..aab6d0a87 100644 --- a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_urls.py +++ b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_urls.py @@ -1,9 +1,9 @@ import pytest -from django.contrib.auth import get_user_model from django.urls import resolve, reverse +from {{ cookiecutter.project_slug }}.users.models import User + pytestmark = pytest.mark.django_db -User = get_user_model() def test_detail(user: User): diff --git a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_views.py b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_views.py index 075cbf8df..59b83b83c 100644 --- a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_views.py +++ b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_views.py @@ -8,6 +8,7 @@ from django.http.response import Http404 from django.test import RequestFactory from {{ cookiecutter.project_slug }}.users.forms import UserChangeForm +from {{ cookiecutter.project_slug }}.users.models import User from {{ cookiecutter.project_slug }}.users.tests.factories import UserFactory from {{ cookiecutter.project_slug }}.users.views import ( UserRedirectView, @@ -16,7 +17,6 @@ from {{ cookiecutter.project_slug }}.users.views import ( ) pytestmark = pytest.mark.django_db -User = get_user_model() class TestUserUpdateView: