Hopiu/django-model-utils
1
0
Fork 0
mirror of https://github.com/Hopiu/django-model-utils.git synced 2026-03-16 20:00:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add minimal annotations to unit tests

Browse source 
These annotations are sufficient to pass mypy inspection if mypy is
configured to allow unannotated functions.
This commit is contained in:
Maarten ter Huurne 2023-03-22 12:42:31 +01:00
parent aeeb69a9dd
commit 218843d754
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/models.py
View file

@ -1,6 +1,6 @@
from __future__ import annotations
from typing import ClassVar
from typing import ClassVar, TypeVar
from django.db import models
from django.db.models import Manager
@ -26,6 +26,8 @@ from model_utils.models import (
from model_utils.tracker import FieldTracker, ModelTracker
from tests.fields import MutableField
ModelT = TypeVar('ModelT', bound=models.Model, covariant=True)
class InheritanceManagerTestRelated(models.Model):
pass
@ -128,7 +130,7 @@ class DoubleMonitored(models.Model):
class Status(StatusModel):
STATUS = Choices(
STATUS: Choices[str] = Choices(
("active", _("active")),
("deleted", _("deleted")),
("on_hold", _("on hold")),
@ -184,7 +186,8 @@ class Post(models.Model):
public: ClassVar[QueryManager[Post]] = QueryManager(published=True)
public_confirmed: ClassVar[QueryManager[Post]] = QueryManager(
models.Q(published=True) & models.Q(confirmed=True))
public_reversed = QueryManager(published=True).order_by("-order")
public_reversed: QueryManager[Post] = QueryManager(
published=True).order_by("-order")
class Meta:
ordering = ("order",)
@ -246,7 +249,7 @@ class TrackedFK(models.Model):
class TrackedAbstract(AbstractTracked):
name = models.CharField(max_length=20)
number = models.IntegerField()
number = models.IntegerField() # type: ignore[assignment]
mutable = MutableField(default=None)
tracker = ModelTracker()