o ]ZhS@sddlZddlZddlZddlmZddlmZddlmZddl Z ddl Z ddl m Z mZGdddZedd ZGd d d Ze jd gd fddZdS)N) literal_eval)contextmanager)dedent)literal_destringizerliteral_stringizerc@seZdZeddZddZddZddZd d Zd d Z e j d de fddZddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(ZdS)) TestGraphcCs d|_dS)Na2Creator "me" Version "xx" graph [ comment "This is a sample graph" directed 1 IsPlanar 1 pos [ x 0 y 1 ] node [ id 1 label "Node 1" pos [ x 1 y 1 ] ] node [ id 2 pos [ x 1 y 2 ] label "Node 2" ] node [ id 3 label "Node 3" pos [ x 1 y 3 ] ] edge [ source 1 target 2 label "Edge from node 1 to node 2" color [line "blue" thickness 3] ] edge [ source 2 target 3 label "Edge from node 2 to node 3" ] edge [ source 3 target 1 label "Edge from node 3 to node 1" ] ] ) simple_data)clsr P/var/www/auris/lib/python3.10/site-packages/networkx/readwrite/tests/test_gml.py setup_classs zTestGraph.setup_classcCsd}t|dS)Nas Creator "Cytoscape" Version 1.0 graph [ node [ root_index -3 id -3 graphics [ x -96.0 y -67.0 w 40.0 h 40.0 fill "#ff9999" type "ellipse" outline "#666666" outline_width 1.5 ] label "node2" ] node [ root_index -2 id -2 graphics [ x 63.0 y 37.0 w 40.0 h 40.0 fill "#ff9999" type "ellipse" outline "#666666" outline_width 1.5 ] label "node1" ] node [ root_index -1 id -1 graphics [ x -31.0 y -17.0 w 40.0 h 40.0 fill "#ff9999" type "ellipse" outline "#666666" outline_width 1.5 ] label "node0" ] edge [ root_index -2 target -2 source -1 graphics [ width 1.5 fill "#0000ff" type "line" Line [ ] source_arrow 0 target_arrow 3 ] label "DirectedEdge" ] edge [ root_index -1 target -1 source -3 graphics [ width 1.5 fill "#0000ff" type "line" Line [ ] source_arrow 0 target_arrow 3 ] label "DirectedEdge" ] ] )nx parse_gml)selfZcytoscape_exampler r r test_parse_gml_cytoscape_bug;sQz&TestGraph.test_parse_gml_cytoscape_bugcCstj|jdd}t|gdksJt|gdks Jt|jddddd d d d d fddddifddddifgksBJdS)NlabelrNode 1Node 2Node 3))rr)rr)rrTdatarrblue)lineZ thicknesszEdge from node 1 to node 2)colorrrzEdge from node 2 to node 3zEdge from node 3 to node 1)r rrsortednodesedgesrGr r r test_parse_gmls  zTestGraph.test_parse_gmlcCs|d}t|d}||jWdn1swYtj|dd}tj|jdd}t|jddt|jddks@Jt|jddt|jddksRJdS)Ntest.gmlwrrTr) openwriterr read_gmlrrrr)rtmp_pathfnamefhZGinr!r r r test_read_gmls $(zTestGraph.test_read_gmlcCs:d}t}|ddtj|td}||ksJdS)Nz0graph [ node [ id 0 label "1203" ] ]i  stringizer)r Graphadd_nodejoin generate_gmlr)ranswerr!rr r r test_labels_are_stringss  z!TestGraph.test_labels_are_stringscCs8d}t|d}|dtjtjtj|dddS)Nz graph [ label "" directed 1 node [ id 0 label "same" ] node [ id 1 label "same" ] ] zUTF-8rrr) ioBytesIOencodeseekpytestraisesr NetworkXErrorr')rrr*r r r test_relabel_duplicates z TestGraph.test_relabel_duplicater.NcCs<t}|dddtj||d}d}||ksJdS)N)r)r=rr,r-zgraph [ node [ id 0 label "(0,1)" ] node [ id 1 label "(1,0)" ] edge [ source 0 target 1 ] ])r r/add_edger1r2)rr.r!rr3r r r test_tuplelabelss  zTestGraph.test_tuplelabelscCstd}d|_dtd}||jdd<t|dd}t|||d| d }Wdn1s;wYd }||ksHJdS) Nr=z path_graph(1)z*This is "quoted" and this is a copyright: rdemor#w+basciizgraph [ name "path_graph(1)" node [ id 0 label "0" demo "This is "quoted" and this is a copyright: ©" ] ]) r path_graphnamechrrr% write_gmlr8readstripdecode)rr(r!attrfobjrr3r r r test_quotess    zTestGraph.test_quotescCsdtd}t}||t|dd}t|||d| d}Wdn1s5wYd}||ksBJdS)Nnoder@r#rBrrCz6graph [ node [ id 0 label "node©" ] ]) rFr r/r0r%rGr8rHrIrJrr(rNr!rLrr3r r r test_unicode_nodes    zTestGraph.test_unicode_nodecCsd}t}||t|dd}t|||d|d}Wdn1s1wYd}||ks>JdS)Ng?r#rBrrCz/graph [ node [ id 0 label "1.0" ] ]) r r/r0r%rGr8rHrIrJrOr r r test_float_labels   zTestGraph.test_float_labelcstdtdtdg}zddl}||j|j|jdg7}Wn ty*||7}Ynwtt|}tt |}t ||dt |j fddt |D}t ||d t|d d ~}t|||d|d }d }||kszJ|dt|} t |D]J\} } | jt| d} t| rt| sJn| | ksJ| } t| dt| df}| j |d }t| rt|sJq|| ksJqWddS1swYdS)Nnanz+infz-infrZ nodefloatcsi|] \}}||qSr r ).0ivaluerr r .z6TestGraph.test_special_float_label..Z edgefloatr#rBrCagraph [ node [ id 0 label "0" nodefloat NAN ] node [ id 1 label "1" nodefloat +INF ] node [ id 2 label "2" nodefloat -INF ] node [ id 3 label "3" nodefloat NAN ] node [ id 4 label "4" nodefloat +INF ] node [ id 5 label "5" nodefloat -INF ] edge [ source 0 target 1 edgefloat NAN ] edge [ source 0 target 5 edgefloat +INF ] edge [ source 1 target 2 edgefloat -INF ] edge [ source 2 target 3 edgefloat NAN ] edge [ source 3 target 4 edgefloat +INF ] edge [ source 4 target 5 edgefloat -INF ] ]r=)floatnumpyrRinf ImportErrorr Z cycle_graphlendict enumerateZset_node_attributeslistrZset_edge_attributesr%rGr8rHrIrJr'rstrmathisnan)rr(Zspecial_floatsnpr!attrsrLrr3graphZindxrVZ node_valueedgeZ string_edgeZ edge_valuer rWr test_special_float_label!sD       >     "z"TestGraph.test_special_float_labelcCsFtd}d|jdksJtd}d|jksJd|jvs!JdS)Nz*graph [ name "x" node [ id 0 label "x" ] ]xrEz!graph [ node [ id 0 label "x" ] ])r rrgrEr r r r test_names  zTestGraph.test_namecCsdD]v}dD]q}d}|dur|dtt|7}|dur&|dtt|7}|d7}|d7}|d7}t|}t||ksAJt||ksKJd}|d urU|d 7}|d ur]|d 7}|d 7}|rg|d 7}|d7}|dt|kswJqqdS)N)NFTzgraph [z directed z multigraph z node [ id 0 label "0" ]z edge [ source 0 target 0 ]z ]zgraph [ Tz directed 1 z multigraph 1 zG node [ id 0 label "0" ] edge [ source 0 target 0 z key 0 z ] ]r,) rbintr rboolZ is_directedZ is_multigraphr1r2)rZdirectedZ multigraphgmlr!r r r test_graph_typess2 zTestGraph.test_graph_typesc CsJddddddddtd d id gg}|td |td t}||_||jd<|jddd|id|jddd|dd tj |t d}tj |t d}||jksVJ||d|jks`Jt|jddddd|idfgkstJt|jddddd|dfgksJt}d|jd<tj t |td}|jddksJdS)NTFl Fx:^VgF"'z"&&&"")DD)r=)ru3iDDz{2.3j, 1 - 2.3j, ()}rrrS)rmrg)rZrr,r-Z destringizer)rErrfrozenset([1, 2, 3]))rFappendrr r/rErgr0r>r1r2rrrrarr)rrr!ror r r test_data_typess2  (& zTestGraph.test_data_typescCsXd}t|}dtdd|jksJdt|}d}d|d}||ks*JdS) NzTgraph [ name "&"䑄��&unknown;" ]z&"rtz*��&unknown;r,z##1234567890;&#x1234567890abcdefz,graph [ name "&"䑄&z;&unknown;" ])r rrFrEr1r2)rror!Zalnur3r r r test_escape_unescapes zTestGraph.test_escape_unescapecCstttdtttdttttttttgdttttt|dd}|tj d| dtt j t j |Wdn1sOwYdd }|d d g|d |d |d|d|d|d|d|d|d|d|d|dt jddd|d|d|d|dt d|d |d!|d"t d#t d$t d%|d&t d't d(t d)|d*d+d,}t }d-|jd-<||t }d-|jd.<||t }tgd|jd/<||td0dS)1N(rxr=rurr#rBsgraph[]rcSsttjtj|dSN)r9r:r r;r)ror r r assert_parse_errorsz5TestGraph.test_exceptions..assert_parse_errorz graph [ ]rkz Creator ""0zgraph ]z graph [ 1 ]zgraph [ 1.E+2 ]z graph [ "A" ]zgraph [ ] graph ]zgraph [ ] graph [ ]zgraph [ data [1, 2, 3] ]zgraph [ node [ ] ]zgraph [ node [ id 0 ] ]zgraph [ node [ id "a" ] ]idrz5graph [ node [ id 0 label 0 ] node [ id 0 label 1 ] ]z5graph [ node [ id 0 label 0 ] node [ id 1 label 0 ] ]z(graph [ node [ id 0 label 0 ] edge [ ] ]z1graph [ node [ id 0 label 0 ] edge [ source 0 ] ]z9graph [edge [ source 0 target 0 ] node [ id 0 label 0 ] ]z:graph [ node [ id 0 label 0 ] edge [ source 1 target 0 ] ]z:graph [ node [ id 0 label 0 ] edge [ source 0 target 1 ] ]zkgraph [ node [ id 0 label 0 ] node [ id 1 label 1 ] edge [ source 0 target 1 ] edge [ source 1 target 0 ] ]zvgraph [ node [ id 0 label 0 ] node [ id 1 label 1 ] edge [ source 0 target 1 ] edge [ source 1 target 0 ] directed 1 ]zwgraph [ node [ id 0 label 0 ] node [ id 1 label 1 ] edge [ source 0 target 1 ] edge [ source 0 target 1 ]multigraph 1 ]z}graph [ node [ id 0 label 0 ] node [ id 1 label 1 ] edge [ source 0 target 1 key 0 ] edge [ source 0 target 1 ]multigraph 1 ]zgraph [ node [ id 0 label 0 ] node [ id 1 label 1 ] edge [ source 0 target 1 key 0 ] edge [ source 0 target 1 key 0 ]multigraph 1 ]zgraph [ node [ id 0 label 0 ] node [ id 1 label 1 ] edge [ source 0 target 1 key 0 ] edge [ source 1 target 0 key 0 ]directed 1 multigraph 1 ]z9graph [edge [ source a target a ] node [ id a label b ] ]zgraph [ node [ id n42 label 0 ] node [ id x43 label 1 ]edge [ source n42 target x43 key 0 ]edge [ source x43 target n42 key 0 ]directed 1 multigraph 1 ]uEgraph [edge [ source '䈀' target '䈀' ] node [ id '䈀' label b ] ]csttjfdddS)NcsttjiSr~)rar r2r argskwargsr r .szJTestGraph.test_exceptions..assert_generate_error..)r9r:r r;rr rr assert_generate_error,sz8TestGraph.test_exceptions..assert_generate_errorrrvrr-)r9r: ValueErrorrr frozensetr%r&codecsBOM_UTF8r8r r;r'rr/rg)rr(frrr!r r r test_exceptionss      zTestGraph.test_exceptionscstj|jddtjgdksJfddtjD}|gdks'Jtj|jddtjgdks:JfddtjD}|gdksNJdS)Nrrr}cg|] }j|dqSrrrTnr!r r >rYz.TestGraph.test_label_kwarg..rcrrrrrr r rCrY)r rrrr)rlabelsr rr test_label_kwarg;szTestGraph.test_label_kwargcCst}dddddddd}|jd i||d }t||t|}|jd D]\}}|d ks8|d krAt|tks@Jq,t|t ksIJq,dS)Nliril)toosmallZsmallZmed1Zmed2Zmed3bigtoobigNoder#rr)r) r r/r0rGr'ritemstyperbrm)rr(r!numbersr)ZG2rKrVr r r test_outofrange_integersFs$   z"TestGraph.test_outofrange_integerscCs*d}t|}|jddddksJdS)Nz graph [ node [ id 0 label "multiline node" label2 "multiline1 multiline2 multiline3" alt_name "id 0" ] ] zmultiline nodez multiline1 multiline2 multiline3zid 0)label2Zalt_name)r rr)rZmultiline_exampler!r r r test_multilineas  zTestGraph.test_multiline)__name__ __module__ __qualname__ classmethodr rr"r+r4r<r9mark parametrizerr?rMrPrQrirlrprzr{rrrrr r r r rs, +U   f ] rccst}|V|ddS)Nr)r5r6r8)Z _file_handler r r byte_filexsrc@s,eZdZddZddZddZddZd S) TestPropertyListscCslt}|jdgddt}t||Wdn1s!wY|}|tdks4JdS)Nn1)elementrr=@TF propertiesa7 graph [ node [ id 0 label "n1" properties "element" properties 0 properties 1 properties 2.5 properties 1 properties 0 ] ] r r/r0rrGrHrJrrgrresultr r r 3test_writing_graph_with_multi_element_property_lists zETestPropertyLists.test_writing_graph_with_multi_element_property_listcCsjt}|jddgdt}t||Wdn1s wY|}|tdks3JdS)Nrrrz graph [ node [ id 0 label "n1" properties "_networkx_list_start" properties "element" ] ] rrr r r 1test_writing_graph_with_one_element_property_lists zCTestPropertyLists.test_writing_graph_with_one_element_property_listcCsrt}|tdd|dt|}Wdn1s"wY|jddddgdiks7JdS) Na graph [ node [ id 0 label "n1" properties "element" properties 0 properties 1 properties 2.5 ] ] rCrTrrr)rrr=rrr&rr7r8r r'rrrrgr r r %test_reading_graph_with_list_propertys   $z7TestPropertyLists.test_reading_graph_with_list_propertycCspt}|tdd|dt|}Wdn1s"wY|jdddddgiks6JdS) Nz graph [ node [ id 0 label "n1" properties "_networkx_list_start" properties "element" ] ] rCrTrrrrrrr r r 4test_reading_graph_with_single_element_list_propertys   "zFTestPropertyLists.test_reading_graph_with_single_element_list_propertyN)rrrrrrrr r r r rs  rcollr cCstd}||jdd<t}t|||dt|}|jdd|ks+Jt|ddd}tj ||s=J|dtj|t d}tj ||sRJdS)Nrurtestrr=)r1rw) r rDrr5r6rGr8r'Z relabel_nodesutilsZ graphs_equalrm)rr!rHr r r test_stringize_empty_list_tuples     r)rr5rcastr contextlibrtextwraprr9Znetworkxr Znetworkx.readwrite.gmlrrrrrrrrr r r r s&   n X