[visitor_test] Test enums

This commit is contained in:
Behdad Esfahbod 2022-08-17 14:27:58 -06:00
parent 2a8f408b31
commit d6bdae1369

View File

@ -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]