diff options
Diffstat (limited to 'tests/postgres_tests/fields.py')
| -rw-r--r-- | tests/postgres_tests/fields.py | 28 |
1 files changed, 19 insertions, 9 deletions
diff --git a/tests/postgres_tests/fields.py b/tests/postgres_tests/fields.py index 1c0cb05d46..2d62e26a92 100644 --- a/tests/postgres_tests/fields.py +++ b/tests/postgres_tests/fields.py @@ -8,31 +8,41 @@ from django.db import models try: from django.contrib.postgres.fields import ( - ArrayField, BigIntegerRangeField, CICharField, CIEmailField, - CITextField, DateRangeField, DateTimeRangeField, DecimalRangeField, - HStoreField, IntegerRangeField, + ArrayField, + BigIntegerRangeField, + CICharField, + CIEmailField, + CITextField, + DateRangeField, + DateTimeRangeField, + DecimalRangeField, + HStoreField, + IntegerRangeField, ) from django.contrib.postgres.search import SearchVector, SearchVectorField except ImportError: + class DummyArrayField(models.Field): def __init__(self, base_field, size=None, **kwargs): super().__init__(**kwargs) def deconstruct(self): name, path, args, kwargs = super().deconstruct() - kwargs.update({ - 'base_field': '', - 'size': 1, - }) + kwargs.update( + { + "base_field": "", + "size": 1, + } + ) return name, path, args, kwargs class DummyContinuousRangeField(models.Field): - def __init__(self, *args, default_bounds='[)', **kwargs): + def __init__(self, *args, default_bounds="[)", **kwargs): super().__init__(**kwargs) def deconstruct(self): name, path, args, kwargs = super().deconstruct() - kwargs['default_bounds'] = '[)' + kwargs["default_bounds"] = "[)" return name, path, args, kwargs ArrayField = DummyArrayField |
