|
1 | 1 | # -*- code: utf8 -*-
|
2 |
| -try: |
3 |
| - import enum |
4 |
| -except ImportError: |
5 |
| - enum = None |
| 2 | +import enum |
6 | 3 | import re
|
7 | 4 | import unittest
|
8 | 5 | from collections import namedtuple
|
@@ -600,36 +597,34 @@ def test_should_fail(self):
|
600 | 597 |
|
601 | 598 | self.assertRaises(ParseError, not_a_digit.parse, "8ab")
|
602 | 599 |
|
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) |
633 | 628 |
|
634 | 629 |
|
635 | 630 | class TestParserTokens(unittest.TestCase):
|
|
0 commit comments