1from sympy.liealgebras.dynkin_diagram import DynkinDiagram
2
3def test_DynkinDiagram():
4    c = DynkinDiagram("A3")
5    diag = "0---0---0\n1   2   3"
6    assert c == diag
7    ct = DynkinDiagram(["B", 3])
8    diag2 = "0---0=>=0\n1   2   3"
9    assert ct == diag2
10