mirror of
https://github.com/django/django.git
synced 2025-10-24 06:06:09 +00:00
Refs #33476 -- Reformatted code with Black.
This commit is contained in:
django-bot
committed by
Mariusz Felisiak
Mariusz Felisiak
parent
f68fa8b45d
commit
9c19aff7c7
@@ -11,47 +11,49 @@ from django.utils.translation import gettext_lazy as _
|
||||
|
||||
|
||||
class Suit(models.IntegerChoices):
|
||||
DIAMOND = 1, _('Diamond')
|
||||
SPADE = 2, _('Spade')
|
||||
HEART = 3, _('Heart')
|
||||
CLUB = 4, _('Club')
|
||||
DIAMOND = 1, _("Diamond")
|
||||
SPADE = 2, _("Spade")
|
||||
HEART = 3, _("Heart")
|
||||
CLUB = 4, _("Club")
|
||||
|
||||
|
||||
class YearInSchool(models.TextChoices):
|
||||
FRESHMAN = 'FR', _('Freshman')
|
||||
SOPHOMORE = 'SO', _('Sophomore')
|
||||
JUNIOR = 'JR', _('Junior')
|
||||
SENIOR = 'SR', _('Senior')
|
||||
GRADUATE = 'GR', _('Graduate')
|
||||
FRESHMAN = "FR", _("Freshman")
|
||||
SOPHOMORE = "SO", _("Sophomore")
|
||||
JUNIOR = "JR", _("Junior")
|
||||
SENIOR = "SR", _("Senior")
|
||||
GRADUATE = "GR", _("Graduate")
|
||||
|
||||
|
||||
class Vehicle(models.IntegerChoices):
|
||||
CAR = 1, 'Carriage'
|
||||
CAR = 1, "Carriage"
|
||||
TRUCK = 2
|
||||
JET_SKI = 3
|
||||
|
||||
__empty__ = _('(Unknown)')
|
||||
__empty__ = _("(Unknown)")
|
||||
|
||||
|
||||
class Gender(models.TextChoices):
|
||||
MALE = 'M'
|
||||
FEMALE = 'F'
|
||||
NOT_SPECIFIED = 'X'
|
||||
MALE = "M"
|
||||
FEMALE = "F"
|
||||
NOT_SPECIFIED = "X"
|
||||
|
||||
__empty__ = '(Undeclared)'
|
||||
__empty__ = "(Undeclared)"
|
||||
|
||||
|
||||
class ChoicesTests(SimpleTestCase):
|
||||
def test_integerchoices(self):
|
||||
self.assertEqual(Suit.choices, [(1, 'Diamond'), (2, 'Spade'), (3, 'Heart'), (4, 'Club')])
|
||||
self.assertEqual(Suit.labels, ['Diamond', 'Spade', 'Heart', 'Club'])
|
||||
self.assertEqual(
|
||||
Suit.choices, [(1, "Diamond"), (2, "Spade"), (3, "Heart"), (4, "Club")]
|
||||
)
|
||||
self.assertEqual(Suit.labels, ["Diamond", "Spade", "Heart", "Club"])
|
||||
self.assertEqual(Suit.values, [1, 2, 3, 4])
|
||||
self.assertEqual(Suit.names, ['DIAMOND', 'SPADE', 'HEART', 'CLUB'])
|
||||
self.assertEqual(Suit.names, ["DIAMOND", "SPADE", "HEART", "CLUB"])
|
||||
|
||||
self.assertEqual(repr(Suit.DIAMOND), 'Suit.DIAMOND')
|
||||
self.assertEqual(Suit.DIAMOND.label, 'Diamond')
|
||||
self.assertEqual(repr(Suit.DIAMOND), "Suit.DIAMOND")
|
||||
self.assertEqual(Suit.DIAMOND.label, "Diamond")
|
||||
self.assertEqual(Suit.DIAMOND.value, 1)
|
||||
self.assertEqual(Suit['DIAMOND'], Suit.DIAMOND)
|
||||
self.assertEqual(Suit["DIAMOND"], Suit.DIAMOND)
|
||||
self.assertEqual(Suit(1), Suit.DIAMOND)
|
||||
|
||||
self.assertIsInstance(Suit, type(models.Choices))
|
||||
@@ -60,21 +62,21 @@ class ChoicesTests(SimpleTestCase):
|
||||
self.assertIsInstance(Suit.DIAMOND.value, int)
|
||||
|
||||
def test_integerchoices_auto_label(self):
|
||||
self.assertEqual(Vehicle.CAR.label, 'Carriage')
|
||||
self.assertEqual(Vehicle.TRUCK.label, 'Truck')
|
||||
self.assertEqual(Vehicle.JET_SKI.label, 'Jet Ski')
|
||||
self.assertEqual(Vehicle.CAR.label, "Carriage")
|
||||
self.assertEqual(Vehicle.TRUCK.label, "Truck")
|
||||
self.assertEqual(Vehicle.JET_SKI.label, "Jet Ski")
|
||||
|
||||
def test_integerchoices_empty_label(self):
|
||||
self.assertEqual(Vehicle.choices[0], (None, '(Unknown)'))
|
||||
self.assertEqual(Vehicle.labels[0], '(Unknown)')
|
||||
self.assertEqual(Vehicle.choices[0], (None, "(Unknown)"))
|
||||
self.assertEqual(Vehicle.labels[0], "(Unknown)")
|
||||
self.assertIsNone(Vehicle.values[0])
|
||||
self.assertEqual(Vehicle.names[0], '__empty__')
|
||||
self.assertEqual(Vehicle.names[0], "__empty__")
|
||||
|
||||
def test_integerchoices_functional_api(self):
|
||||
Place = models.IntegerChoices('Place', 'FIRST SECOND THIRD')
|
||||
self.assertEqual(Place.labels, ['First', 'Second', 'Third'])
|
||||
Place = models.IntegerChoices("Place", "FIRST SECOND THIRD")
|
||||
self.assertEqual(Place.labels, ["First", "Second", "Third"])
|
||||
self.assertEqual(Place.values, [1, 2, 3])
|
||||
self.assertEqual(Place.names, ['FIRST', 'SECOND', 'THIRD'])
|
||||
self.assertEqual(Place.names, ["FIRST", "SECOND", "THIRD"])
|
||||
|
||||
def test_integerchoices_containment(self):
|
||||
self.assertIn(Suit.DIAMOND, Suit)
|
||||
@@ -82,18 +84,31 @@ class ChoicesTests(SimpleTestCase):
|
||||
self.assertNotIn(0, Suit)
|
||||
|
||||
def test_textchoices(self):
|
||||
self.assertEqual(YearInSchool.choices, [
|
||||
('FR', 'Freshman'), ('SO', 'Sophomore'), ('JR', 'Junior'), ('SR', 'Senior'), ('GR', 'Graduate'),
|
||||
])
|
||||
self.assertEqual(YearInSchool.labels, ['Freshman', 'Sophomore', 'Junior', 'Senior', 'Graduate'])
|
||||
self.assertEqual(YearInSchool.values, ['FR', 'SO', 'JR', 'SR', 'GR'])
|
||||
self.assertEqual(YearInSchool.names, ['FRESHMAN', 'SOPHOMORE', 'JUNIOR', 'SENIOR', 'GRADUATE'])
|
||||
self.assertEqual(
|
||||
YearInSchool.choices,
|
||||
[
|
||||
("FR", "Freshman"),
|
||||
("SO", "Sophomore"),
|
||||
("JR", "Junior"),
|
||||
("SR", "Senior"),
|
||||
("GR", "Graduate"),
|
||||
],
|
||||
)
|
||||
self.assertEqual(
|
||||
YearInSchool.labels,
|
||||
["Freshman", "Sophomore", "Junior", "Senior", "Graduate"],
|
||||
)
|
||||
self.assertEqual(YearInSchool.values, ["FR", "SO", "JR", "SR", "GR"])
|
||||
self.assertEqual(
|
||||
YearInSchool.names,
|
||||
["FRESHMAN", "SOPHOMORE", "JUNIOR", "SENIOR", "GRADUATE"],
|
||||
)
|
||||
|
||||
self.assertEqual(repr(YearInSchool.FRESHMAN), 'YearInSchool.FRESHMAN')
|
||||
self.assertEqual(YearInSchool.FRESHMAN.label, 'Freshman')
|
||||
self.assertEqual(YearInSchool.FRESHMAN.value, 'FR')
|
||||
self.assertEqual(YearInSchool['FRESHMAN'], YearInSchool.FRESHMAN)
|
||||
self.assertEqual(YearInSchool('FR'), YearInSchool.FRESHMAN)
|
||||
self.assertEqual(repr(YearInSchool.FRESHMAN), "YearInSchool.FRESHMAN")
|
||||
self.assertEqual(YearInSchool.FRESHMAN.label, "Freshman")
|
||||
self.assertEqual(YearInSchool.FRESHMAN.value, "FR")
|
||||
self.assertEqual(YearInSchool["FRESHMAN"], YearInSchool.FRESHMAN)
|
||||
self.assertEqual(YearInSchool("FR"), YearInSchool.FRESHMAN)
|
||||
|
||||
self.assertIsInstance(YearInSchool, type(models.Choices))
|
||||
self.assertIsInstance(YearInSchool.FRESHMAN, YearInSchool)
|
||||
@@ -101,48 +116,50 @@ class ChoicesTests(SimpleTestCase):
|
||||
self.assertIsInstance(YearInSchool.FRESHMAN.value, str)
|
||||
|
||||
def test_textchoices_auto_label(self):
|
||||
self.assertEqual(Gender.MALE.label, 'Male')
|
||||
self.assertEqual(Gender.FEMALE.label, 'Female')
|
||||
self.assertEqual(Gender.NOT_SPECIFIED.label, 'Not Specified')
|
||||
self.assertEqual(Gender.MALE.label, "Male")
|
||||
self.assertEqual(Gender.FEMALE.label, "Female")
|
||||
self.assertEqual(Gender.NOT_SPECIFIED.label, "Not Specified")
|
||||
|
||||
def test_textchoices_empty_label(self):
|
||||
self.assertEqual(Gender.choices[0], (None, '(Undeclared)'))
|
||||
self.assertEqual(Gender.labels[0], '(Undeclared)')
|
||||
self.assertEqual(Gender.choices[0], (None, "(Undeclared)"))
|
||||
self.assertEqual(Gender.labels[0], "(Undeclared)")
|
||||
self.assertIsNone(Gender.values[0])
|
||||
self.assertEqual(Gender.names[0], '__empty__')
|
||||
self.assertEqual(Gender.names[0], "__empty__")
|
||||
|
||||
def test_textchoices_functional_api(self):
|
||||
Medal = models.TextChoices('Medal', 'GOLD SILVER BRONZE')
|
||||
self.assertEqual(Medal.labels, ['Gold', 'Silver', 'Bronze'])
|
||||
self.assertEqual(Medal.values, ['GOLD', 'SILVER', 'BRONZE'])
|
||||
self.assertEqual(Medal.names, ['GOLD', 'SILVER', 'BRONZE'])
|
||||
Medal = models.TextChoices("Medal", "GOLD SILVER BRONZE")
|
||||
self.assertEqual(Medal.labels, ["Gold", "Silver", "Bronze"])
|
||||
self.assertEqual(Medal.values, ["GOLD", "SILVER", "BRONZE"])
|
||||
self.assertEqual(Medal.names, ["GOLD", "SILVER", "BRONZE"])
|
||||
|
||||
def test_textchoices_containment(self):
|
||||
self.assertIn(YearInSchool.FRESHMAN, YearInSchool)
|
||||
self.assertIn('FR', YearInSchool)
|
||||
self.assertNotIn('XX', YearInSchool)
|
||||
self.assertIn("FR", YearInSchool)
|
||||
self.assertNotIn("XX", YearInSchool)
|
||||
|
||||
def test_textchoices_blank_value(self):
|
||||
class BlankStr(models.TextChoices):
|
||||
EMPTY = '', '(Empty)'
|
||||
ONE = 'ONE', 'One'
|
||||
EMPTY = "", "(Empty)"
|
||||
ONE = "ONE", "One"
|
||||
|
||||
self.assertEqual(BlankStr.labels, ['(Empty)', 'One'])
|
||||
self.assertEqual(BlankStr.values, ['', 'ONE'])
|
||||
self.assertEqual(BlankStr.names, ['EMPTY', 'ONE'])
|
||||
self.assertEqual(BlankStr.labels, ["(Empty)", "One"])
|
||||
self.assertEqual(BlankStr.values, ["", "ONE"])
|
||||
self.assertEqual(BlankStr.names, ["EMPTY", "ONE"])
|
||||
|
||||
def test_invalid_definition(self):
|
||||
msg = "'str' object cannot be interpreted as an integer"
|
||||
with self.assertRaisesMessage(TypeError, msg):
|
||||
|
||||
class InvalidArgumentEnum(models.IntegerChoices):
|
||||
# A string is not permitted as the second argument to int().
|
||||
ONE = 1, 'X', 'Invalid'
|
||||
ONE = 1, "X", "Invalid"
|
||||
|
||||
msg = "duplicate values found in <enum 'Fruit'>: PINEAPPLE -> APPLE"
|
||||
with self.assertRaisesMessage(ValueError, msg):
|
||||
|
||||
class Fruit(models.IntegerChoices):
|
||||
APPLE = 1, 'Apple'
|
||||
PINEAPPLE = 1, 'Pineapple'
|
||||
APPLE = 1, "Apple"
|
||||
PINEAPPLE = 1, "Pineapple"
|
||||
|
||||
def test_str(self):
|
||||
for test in [Gender, Suit, YearInSchool, Vehicle]:
|
||||
@@ -151,45 +168,45 @@ class ChoicesTests(SimpleTestCase):
|
||||
self.assertEqual(str(test[member.name]), str(member.value))
|
||||
|
||||
def test_templates(self):
|
||||
template = Template('{{ Suit.DIAMOND.label }}|{{ Suit.DIAMOND.value }}')
|
||||
output = template.render(Context({'Suit': Suit}))
|
||||
self.assertEqual(output, 'Diamond|1')
|
||||
template = Template("{{ Suit.DIAMOND.label }}|{{ Suit.DIAMOND.value }}")
|
||||
output = template.render(Context({"Suit": Suit}))
|
||||
self.assertEqual(output, "Diamond|1")
|
||||
|
||||
def test_property_names_conflict_with_member_names(self):
|
||||
with self.assertRaises(AttributeError):
|
||||
models.TextChoices('Properties', 'choices labels names values')
|
||||
models.TextChoices("Properties", "choices labels names values")
|
||||
|
||||
def test_label_member(self):
|
||||
# label can be used as a member.
|
||||
Stationery = models.TextChoices('Stationery', 'label stamp sticker')
|
||||
self.assertEqual(Stationery.label.label, 'Label')
|
||||
self.assertEqual(Stationery.label.value, 'label')
|
||||
self.assertEqual(Stationery.label.name, 'label')
|
||||
Stationery = models.TextChoices("Stationery", "label stamp sticker")
|
||||
self.assertEqual(Stationery.label.label, "Label")
|
||||
self.assertEqual(Stationery.label.value, "label")
|
||||
self.assertEqual(Stationery.label.name, "label")
|
||||
|
||||
def test_do_not_call_in_templates_member(self):
|
||||
# do_not_call_in_templates is not implicitly treated as a member.
|
||||
Special = models.IntegerChoices('Special', 'do_not_call_in_templates')
|
||||
Special = models.IntegerChoices("Special", "do_not_call_in_templates")
|
||||
self.assertEqual(
|
||||
Special.do_not_call_in_templates.label,
|
||||
'Do Not Call In Templates',
|
||||
"Do Not Call In Templates",
|
||||
)
|
||||
self.assertEqual(Special.do_not_call_in_templates.value, 1)
|
||||
self.assertEqual(
|
||||
Special.do_not_call_in_templates.name,
|
||||
'do_not_call_in_templates',
|
||||
"do_not_call_in_templates",
|
||||
)
|
||||
|
||||
|
||||
class Separator(bytes, models.Choices):
|
||||
FS = b'\x1c', 'File Separator'
|
||||
GS = b'\x1d', 'Group Separator'
|
||||
RS = b'\x1e', 'Record Separator'
|
||||
US = b'\x1f', 'Unit Separator'
|
||||
FS = b"\x1c", "File Separator"
|
||||
GS = b"\x1d", "Group Separator"
|
||||
RS = b"\x1e", "Record Separator"
|
||||
US = b"\x1f", "Unit Separator"
|
||||
|
||||
|
||||
class Constants(float, models.Choices):
|
||||
PI = 3.141592653589793, 'π'
|
||||
TAU = 6.283185307179586, 'τ'
|
||||
PI = 3.141592653589793, "π"
|
||||
TAU = 6.283185307179586, "τ"
|
||||
|
||||
|
||||
class Set(frozenset, models.Choices):
|
||||
@@ -201,12 +218,12 @@ class Set(frozenset, models.Choices):
|
||||
|
||||
|
||||
class MoonLandings(datetime.date, models.Choices):
|
||||
APOLLO_11 = 1969, 7, 20, 'Apollo 11 (Eagle)'
|
||||
APOLLO_12 = 1969, 11, 19, 'Apollo 12 (Intrepid)'
|
||||
APOLLO_14 = 1971, 2, 5, 'Apollo 14 (Antares)'
|
||||
APOLLO_15 = 1971, 7, 30, 'Apollo 15 (Falcon)'
|
||||
APOLLO_16 = 1972, 4, 21, 'Apollo 16 (Orion)'
|
||||
APOLLO_17 = 1972, 12, 11, 'Apollo 17 (Challenger)'
|
||||
APOLLO_11 = 1969, 7, 20, "Apollo 11 (Eagle)"
|
||||
APOLLO_12 = 1969, 11, 19, "Apollo 12 (Intrepid)"
|
||||
APOLLO_14 = 1971, 2, 5, "Apollo 14 (Antares)"
|
||||
APOLLO_15 = 1971, 7, 30, "Apollo 15 (Falcon)"
|
||||
APOLLO_16 = 1972, 4, 21, "Apollo 16 (Orion)"
|
||||
APOLLO_17 = 1972, 12, 11, "Apollo 17 (Challenger)"
|
||||
|
||||
|
||||
class DateAndTime(datetime.datetime, models.Choices):
|
||||
@@ -222,48 +239,57 @@ class MealTimes(datetime.time, models.Choices):
|
||||
|
||||
|
||||
class Frequency(datetime.timedelta, models.Choices):
|
||||
WEEK = 0, 0, 0, 0, 0, 0, 1, 'Week'
|
||||
DAY = 1, 'Day'
|
||||
HOUR = 0, 0, 0, 0, 0, 1, 'Hour'
|
||||
MINUTE = 0, 0, 0, 0, 1, 'Hour'
|
||||
SECOND = 0, 1, 'Second'
|
||||
WEEK = 0, 0, 0, 0, 0, 0, 1, "Week"
|
||||
DAY = 1, "Day"
|
||||
HOUR = 0, 0, 0, 0, 0, 1, "Hour"
|
||||
MINUTE = 0, 0, 0, 0, 1, "Hour"
|
||||
SECOND = 0, 1, "Second"
|
||||
|
||||
|
||||
class Number(decimal.Decimal, models.Choices):
|
||||
E = 2.718281828459045, 'e'
|
||||
PI = '3.141592653589793', 'π'
|
||||
TAU = decimal.Decimal('6.283185307179586'), 'τ'
|
||||
E = 2.718281828459045, "e"
|
||||
PI = "3.141592653589793", "π"
|
||||
TAU = decimal.Decimal("6.283185307179586"), "τ"
|
||||
|
||||
|
||||
class IPv4Address(ipaddress.IPv4Address, models.Choices):
|
||||
LOCALHOST = '127.0.0.1', 'Localhost'
|
||||
GATEWAY = '192.168.0.1', 'Gateway'
|
||||
BROADCAST = '192.168.0.255', 'Broadcast'
|
||||
LOCALHOST = "127.0.0.1", "Localhost"
|
||||
GATEWAY = "192.168.0.1", "Gateway"
|
||||
BROADCAST = "192.168.0.255", "Broadcast"
|
||||
|
||||
|
||||
class IPv6Address(ipaddress.IPv6Address, models.Choices):
|
||||
LOCALHOST = '::1', 'Localhost'
|
||||
UNSPECIFIED = '::', 'Unspecified'
|
||||
LOCALHOST = "::1", "Localhost"
|
||||
UNSPECIFIED = "::", "Unspecified"
|
||||
|
||||
|
||||
class IPv4Network(ipaddress.IPv4Network, models.Choices):
|
||||
LOOPBACK = '127.0.0.0/8', 'Loopback'
|
||||
LINK_LOCAL = '169.254.0.0/16', 'Link-Local'
|
||||
PRIVATE_USE_A = '10.0.0.0/8', 'Private-Use (Class A)'
|
||||
LOOPBACK = "127.0.0.0/8", "Loopback"
|
||||
LINK_LOCAL = "169.254.0.0/16", "Link-Local"
|
||||
PRIVATE_USE_A = "10.0.0.0/8", "Private-Use (Class A)"
|
||||
|
||||
|
||||
class IPv6Network(ipaddress.IPv6Network, models.Choices):
|
||||
LOOPBACK = '::1/128', 'Loopback'
|
||||
UNSPECIFIED = '::/128', 'Unspecified'
|
||||
UNIQUE_LOCAL = 'fc00::/7', 'Unique-Local'
|
||||
LINK_LOCAL_UNICAST = 'fe80::/10', 'Link-Local Unicast'
|
||||
LOOPBACK = "::1/128", "Loopback"
|
||||
UNSPECIFIED = "::/128", "Unspecified"
|
||||
UNIQUE_LOCAL = "fc00::/7", "Unique-Local"
|
||||
LINK_LOCAL_UNICAST = "fe80::/10", "Link-Local Unicast"
|
||||
|
||||
|
||||
class CustomChoicesTests(SimpleTestCase):
|
||||
def test_labels_valid(self):
|
||||
enums = (
|
||||
Separator, Constants, Set, MoonLandings, DateAndTime, MealTimes,
|
||||
Frequency, Number, IPv4Address, IPv6Address, IPv4Network,
|
||||
Separator,
|
||||
Constants,
|
||||
Set,
|
||||
MoonLandings,
|
||||
DateAndTime,
|
||||
MealTimes,
|
||||
Frequency,
|
||||
Number,
|
||||
IPv4Address,
|
||||
IPv6Address,
|
||||
IPv4Network,
|
||||
IPv6Network,
|
||||
)
|
||||
for choice_enum in enums:
|
||||
@@ -273,17 +299,20 @@ class CustomChoicesTests(SimpleTestCase):
|
||||
def test_bool_unsupported(self):
|
||||
msg = "type 'bool' is not an acceptable base type"
|
||||
with self.assertRaisesMessage(TypeError, msg):
|
||||
|
||||
class Boolean(bool, models.Choices):
|
||||
pass
|
||||
|
||||
def test_timezone_unsupported(self):
|
||||
msg = "type 'datetime.timezone' is not an acceptable base type"
|
||||
with self.assertRaisesMessage(TypeError, msg):
|
||||
|
||||
class Timezone(datetime.timezone, models.Choices):
|
||||
pass
|
||||
|
||||
def test_uuid_unsupported(self):
|
||||
msg = 'UUID objects are immutable'
|
||||
msg = "UUID objects are immutable"
|
||||
with self.assertRaisesMessage(TypeError, msg):
|
||||
|
||||
class Identifier(uuid.UUID, models.Choices):
|
||||
A = '972ce4eb-a95f-4a56-9339-68c208a76f18'
|
||||
A = "972ce4eb-a95f-4a56-9339-68c208a76f18"
|
||||
|
||||
Reference in New Issue
Block a user