o ˷e@sXddlmZddlZddlZddZddZddZd d Zd d Z d dZ ddZ dS))dedentNcCstjddtjd}|jD]}dttd||j|d<q td}td}tj|dd }t |||ks;Jtj|d d }t |||ksLJg}tj||j dd }|dus]J|| d ksfJtj|t d }|dussJdS)Nrh create_usingnode_alabelu ╙── 0 ├─╼ 1 │   ├─╼ 3 │   └─╼ 4 └─╼ 2 ├─╼ 5 └─╼ 6 u ╙── node_a ├─╼ node_b │   ├─╼ node_d │   └─╼ node_e └─╼ node_c ├─╼ node_f └─╼ node_g F) with_labelsT)writer  r ) nx balanced_treeDiGraphnodeschrordrstrip forest_strprintappendsplit)graphnode node_target label_targetretlinesr Y/var/www/ideatree/venv/lib/python3.10/site-packages/networkx/readwrite/tests/test_text.pytest_directed_tree_strs2         r"cCs0ttdks JttdksJdS)Nu╙)rrrGraphr r r r!test_empty_graph=sr$cCstjddtjd}tjddtjd}t||g}t|}t|td}||ks.Jtjddtjd}t|||g}tj|gdd}t|td}||ksXJtj|gddd}t|td }||ksrJdS) Nrru ╟── 0 ╎   ├─╼ 1 ╎   │   ├─╼ 3 ╎   │   └─╼ 4 ╎   └─╼ 2 ╎   ├─╼ 5 ╎   └─╼ 6 ╙── 7 ├─╼ 8 │   ├─╼ 10 │   └─╼ 11 └─╼ 9 ├─╼ 12 └─╼ 13 )rsourcesux ╟── 0 ╎   ├─╼ 1 ╎   │   ├─╼ 3 ╎   │   └─╼ 4 ╎   └─╼ 2 ╎   ├─╼ 5 ╎   └─╼ 6 ╟── 14 ╎   ├─╼ 15 ╎   │   ├─╼ 17 ╎   │   └─╼ 18 ╎   └─╼ 16 ╎   ├─╼ 19 ╎   └─╼ 20 ╙── 7 ├─╼ 8 │   ├─╼ 10 │   └─╼ 11 └─╼ 9 ├─╼ 12 └─╼ 13 Tr( ascii_onlyu +-- 0 :   |-> 1 :   |   |-> 3 :   |   L-> 4 :   L-> 2 :   |-> 5 :   L-> 6 +-- 14 :   |-> 15 :   |   |-> 17 :   |   L-> 18 :   L-> 16 :   |-> 19 :   L-> 20 +-- 7 |-> 8 |   |-> 10 |   L-> 11 L-> 9 |-> 12 L-> 13 )rrrdisjoint_union_allrrrr)tree1tree2forestrtargettree3r r r!test_directed_multi_tree_forestBs:   r1cstjddtjdtjddtjd}t|fdd|jD}t|}tj|ddgd}t|td }||ks?Jtj|ddgd d }t|td  }||ksYJdS) Nrrcsi|] }||tqSr )len).0nr,r r! sz5test_undirected_multi_tree_forest..rr&r'u ╟── 0 ╎   ├── 1 ╎   │   ├── 3 ╎   │   └── 4 ╎   └── 2 ╎   ├── 5 ╎   └── 6 ╙── 7 ├── 8 │   ├── 10 │   └── 11 └── 9 ├── 12 └── 13 Tr)u5 +-- 0 :   |-- 1 :   |   |-- 3 :   |   L-- 4 :   L-- 2 :   |-- 5 :   L-- 6 +-- 7 |-- 8 |   |-- 10 |   L-- 11 L-- 9 |-- 12 L-- 13 ) rrr# relabel_nodesrunionrrrr)r-r.rr/r r5r!!test_undirected_multi_tree_forests(  r9cCstjddtjd}t|td}tj|dgd}t|||ks&Jtd}tj|dgd}t|||ks>JdS)Nrru ╙── 0 ├── 1 │   ├── 3 │   └── 4 └── 2 ├── 5 └── 6 rr'u ╙── 2 ├── 0 │   └── 1 │   ├── 3 │   └── 4 ├── 5 └── 6 )rrr#rrrr)r node_target0r node_target2r r r!test_undirected_tree_strs$      r<cCstjdtjd}ttj t|Wdn1swYtjdtjd}ttjt|WddS1sBwYdS)N)r)rcomplete_graphr#pytestraisesNetworkXNotImplementedrr)ugraphdgraphr r r!test_forest_str_errorss  "rDc Csttjddtjdtjddtjdtjddtjdg}td}td}g}tj||j|jdt d |}t dt |g}tj||jd t d |}t d t |||kscJ||ksiJd S) z When sources are directly specified, we wont be able to determine when we are in the last component, so there will always be a trailing, leftmost pipe. rru ╟── 0 ╎   ├─╼ 1 ╎   └─╼ 2 ╟── 3 ╎   └─╼ 4 ╎   └─╼ 5 ╟── 6 ╎   ├─╼ 7 ╎   └─╼ 8 u ╟── 0 ╎   ├─╼ 1 ╎   └─╼ 2 ╟── 3 ╎   └─╼ 4 ╎   └─╼ 5 ╙── 6 ├─╼ 7 └─╼ 8 )r r( zgot1: rzgot2: N) rr+rrrrrrrrjoinr)rtarget1target2rgot1got2r r r!test_overspecified_sourcess8    rL) textwraprr?networkxrr"r$r1r9r<rDrLr r r r!s 5Z4)