Files
django-stubs/test/pytest_tests/test_postgres_fields.py
Maxim Kurnikov f59cfe6371 latest changes
2018-11-26 23:58:34 +03:00

28 lines
1.0 KiB
Python

from test.pytest_plugin import reveal_type
from test.pytest_tests.base import BaseDjangoPluginTestCase
class TestArrayField(BaseDjangoPluginTestCase):
def test_descriptor_access(self):
from django.db import models
from django.contrib.postgres.fields import ArrayField
class User(models.Model):
array = ArrayField(base_field=models.Field())
user = User()
reveal_type(user.array) # E: Revealed type is 'builtins.list[Any]'
def test_base_field_parsed_into_generic_attribute(self):
from django.db import models
from django.contrib.postgres.fields import ArrayField
class User(models.Model):
members = ArrayField(base_field=models.IntegerField())
members_as_text = ArrayField(base_field=models.CharField(max_length=255))
user = User()
reveal_type(user.members) # E: Revealed type is 'builtins.list[builtins.int*]'
reveal_type(user.members_as_text) # E: Revealed type is 'builtins.list[builtins.str*]'