Skip to content

Commit 590a857

Browse files
committed
Dropped conditions around 'enum` module
It's available in all versions we support
1 parent 1c6ca97 commit 590a857

File tree

1 file changed

+29
-34
lines changed

1 file changed

+29
-34
lines changed

tests/test_parsy.py

Lines changed: 29 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,5 @@
11
# -*- code: utf8 -*-
2-
try:
3-
import enum
4-
except ImportError:
5-
enum = None
2+
import enum
63
import re
74
import unittest
85
from collections import namedtuple
@@ -600,36 +597,34 @@ def test_should_fail(self):
600597

601598
self.assertRaises(ParseError, not_a_digit.parse, "8ab")
602599

603-
if enum is not None:
604-
605-
def test_from_enum_string(self):
606-
class Pet(enum.Enum):
607-
CAT = "cat"
608-
DOG = "dog"
609-
610-
pet = from_enum(Pet)
611-
self.assertEqual(pet.parse("cat"), Pet.CAT)
612-
self.assertEqual(pet.parse("dog"), Pet.DOG)
613-
self.assertRaises(ParseError, pet.parse, "foo")
614-
615-
def test_from_enum_int(self):
616-
class Position(enum.Enum):
617-
FIRST = 1
618-
SECOND = 2
619-
620-
position = from_enum(Position)
621-
self.assertEqual(position.parse("1"), Position.FIRST)
622-
self.assertEqual(position.parse("2"), Position.SECOND)
623-
self.assertRaises(ParseError, position.parse, "foo")
624-
625-
def test_from_enum_transform(self):
626-
class Pet(enum.Enum):
627-
CAT = "cat"
628-
DOG = "dog"
629-
630-
pet = from_enum(Pet, transform=lambda s: s.lower())
631-
self.assertEqual(pet.parse("cat"), Pet.CAT)
632-
self.assertEqual(pet.parse("CAT"), Pet.CAT)
600+
def test_from_enum_string(self):
601+
class Pet(enum.Enum):
602+
CAT = "cat"
603+
DOG = "dog"
604+
605+
pet = from_enum(Pet)
606+
self.assertEqual(pet.parse("cat"), Pet.CAT)
607+
self.assertEqual(pet.parse("dog"), Pet.DOG)
608+
self.assertRaises(ParseError, pet.parse, "foo")
609+
610+
def test_from_enum_int(self):
611+
class Position(enum.Enum):
612+
FIRST = 1
613+
SECOND = 2
614+
615+
position = from_enum(Position)
616+
self.assertEqual(position.parse("1"), Position.FIRST)
617+
self.assertEqual(position.parse("2"), Position.SECOND)
618+
self.assertRaises(ParseError, position.parse, "foo")
619+
620+
def test_from_enum_transform(self):
621+
class Pet(enum.Enum):
622+
CAT = "cat"
623+
DOG = "dog"
624+
625+
pet = from_enum(Pet, transform=lambda s: s.lower())
626+
self.assertEqual(pet.parse("cat"), Pet.CAT)
627+
self.assertEqual(pet.parse("CAT"), Pet.CAT)
633628

634629

635630
class TestParserTokens(unittest.TestCase):

0 commit comments

Comments
 (0)