latest changes

test/pytest_tests/test_postgres_fields.py

@@ -0,0 +1,27 @@
+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*]'
+