Function test_reference_cycle_custom_repr

tests/test_pretty.py:393–417

()

def test_reference_cycle_custom_repr() -> None:
    class Example:
        def __init__(self, x, y):
            self.x = x
            self.y = y

        def __rich_repr__(self):
            yield (class="st">"x", self.x)
            yield (class="st">"y", self.y)

    test = Example(1, None)
    test.y = test
    res = pretty_repr(test)
    assert res == class="st">"Example(x=1, y=...)"

    test = Example(1, Example(2, None))
    test.y.y = test
    res = pretty_repr(test)
    assert res == class="st">"Example(x=1, y=Example(x=2, y=...))"

    class="cm"># Not a cyclic reference, just a repeated reference
    a = Example(2, None)
    test = Example(1, [a, a])
    res = pretty_repr(test)
    assert res == class="st">"Example(x=1, y=[Example(x=2, y=None), Example(x=2, y=None)])"


def test_max_depth() -> None: