[visitor_test] Test enums
This commit is contained in:
parent
2a8f408b31
commit
d6bdae1369
@ -1,14 +1,20 @@
|
|||||||
from fontTools.misc.visitor import Visitor
|
from fontTools.misc.visitor import Visitor
|
||||||
|
import enum
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
|
class E(enum.Enum):
|
||||||
|
E1 = 1
|
||||||
|
E2 = 2
|
||||||
|
E3 = 3
|
||||||
|
|
||||||
class A:
|
class A:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.a = 1
|
self.a = 1
|
||||||
self.b = [2, 3]
|
self.b = [2, 3]
|
||||||
self.c = {4: 5, 6: 7}
|
self.c = {4: 5, 6: 7}
|
||||||
self._d = 8
|
self._d = 8
|
||||||
self.e = 9
|
self.e = E.E2
|
||||||
self.f = 10
|
self.f = 10
|
||||||
|
|
||||||
|
|
||||||
@ -58,4 +64,4 @@ class VisitorTest(object):
|
|||||||
b = B()
|
b = B()
|
||||||
visitor = TestVisitor()
|
visitor = TestVisitor()
|
||||||
visitor.visit(b)
|
visitor.visit(b)
|
||||||
assert visitor.value == ["B", "B a", "A", 1, 2, 3, 5, 7, "e", 9, 10]
|
assert visitor.value == ["B", "B a", "A", 1, 2, 3, 5, 7, "e", E.E2, 10]
|
||||||
|
Loading…
x
Reference in New Issue
Block a user