simplify the pytz import as it should always be available

This commit is contained in:
Jakub Kulík 2025-07-21 14:13:31 +02:00
parent a415b604e9
commit ecccc156ed

View File

@ -10,11 +10,7 @@ from unittest.mock import patch
from zoneinfo import ZoneInfo from zoneinfo import ZoneInfo
import pytest import pytest
import pytz
try:
import pytz
except ImportError:
pytz = None
import django import django
from django.core.exceptions import ValidationError as DjangoValidationError from django.core.exceptions import ValidationError as DjangoValidationError
@ -1642,16 +1638,15 @@ class TestPytzNaiveDayLightSavingTimeTimeZoneDateTimeField(FieldValues):
} }
outputs = {} outputs = {}
if pytz: class MockTimezone(pytz.BaseTzInfo):
class MockTimezone(pytz.BaseTzInfo): @staticmethod
@staticmethod def localize(value, is_dst):
def localize(value, is_dst): raise pytz.InvalidTimeError()
raise pytz.InvalidTimeError()
def __str__(self): def __str__(self):
return 'America/New_York' return 'America/New_York'
field = serializers.DateTimeField(default_timezone=MockTimezone()) field = serializers.DateTimeField(default_timezone=MockTimezone())
@patch('rest_framework.utils.timezone.datetime_ambiguous', return_value=True) @patch('rest_framework.utils.timezone.datetime_ambiguous', return_value=True)