ai-content-maker/.venv/Lib/site-packages/sympy/liealgebras/tests/test_dynkin_diagram.py

from sympy.liealgebras.dynkin_diagram import DynkinDiagram

def test_DynkinDiagram():
    c = DynkinDiagram("A3")
    diag = "0---0---0\n1   2   3"
    assert c == diag
    ct = DynkinDiagram(["B", 3])
    diag2 = "0---0=>=0\n1   2   3"
    assert ct == diag2