From f676472ff345a647eb3b62026e3829b4da2adff0 Mon Sep 17 00:00:00 2001 From: Simon Cozens Date: Wed, 3 Jul 2024 13:44:37 +0100 Subject: [PATCH] Test for visitor inheritance --- Tests/misc/visitor_test.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/Tests/misc/visitor_test.py b/Tests/misc/visitor_test.py index 268cc716d..eb1969ea8 100644 --- a/Tests/misc/visitor_test.py +++ b/Tests/misc/visitor_test.py @@ -24,6 +24,10 @@ class B: self.a = A() +class C(B): + pass + + class TestVisitor(Visitor): def __init__(self): self.value = [] @@ -71,3 +75,9 @@ class VisitorTest(object): visitor.defaultStop = True visitor.visit(b) assert visitor.value == ["B", "B a"] + + def test_visitor_inheritance(self): + b = C() # Should behave just like a B() + visitor = TestVisitor() + visitor.visit(b) + assert visitor.value == ["B", "B a", "A", 1, 2, 3, 5, 7, "e", E.E2, 10]