a hVS@sddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl Z ddl Z ddlmZmZGdddZedd 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 O/var/www/auris/lib/python3.9/site-packages/networkx/readwrite/tests/test_gml.py setup_classszTestGraph.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|gdks(Jt|gdks@Jt|jddddd d d d d fddddifddddifgksJdS)NlabelrNode 1Node 2Node 3))rr)rr)rrTdatarrZblue)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_gmlcCst\}}t|d}||j|tj|dd}tj|jdd}t |j ddt |j ddkslJt |j ddt |j ddksJt |t |dS)NwrrTr)tempfilemkstempopenwritercloser read_gmlrrrrosunlink)rfdfnamefhZGinr r r r test_read_gmls   $$ zTestGraph.test_read_gmlcCs:d}t}|ddtj|td}||ks6JdS)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_duplicater1NcCs<t}|dddtj||d}d}||ks8JdS)N)r)r@rr/r0zgraph [ node [ id 0 label "(0,1)" ] node [ id 1 label "(1,0)" ] edge [ source 0 target 1 ] ])r r2add_edger4r5)rr1r rr6r r r test_tuplelabelss  zTestGraph.test_tuplelabelscCsntd}d|_dtd}||jdd<t}t|||d| d}d}||ksjJdS) Nr@z path_graph(1)z*This is "quoted" and this is a copyright: rdemoasciizgraph [ name "path_graph(1)" node [ id 0 label "0" demo "This is "quoted" and this is a copyright: ©" ] ]) r path_graphnamechrrr#NamedTemporaryFile write_gmlr;readstripdecode)rr attrfobjrr6r r r test_quotess    zTestGraph.test_quotescCsbdtd}t}||t}t|||d| d}d}||ks^JdS)NnoderCrrEz6graph [ node [ id 0 label "node©" ] ]) rHr r2r3r#rIrJr;rKrLrMrrQr rOrr6r r r test_unicode_nodes    zTestGraph.test_unicode_nodecCsZd}t}||t}t|||d| d}d}||ksVJdS)Ng?rrEz/graph [ node [ id 0 label "1.0" ] ]) r r2r3r#rIrJr;rKrLrMrRr r r test_float_labels   zTestGraph.test_float_labelcstdtdtdg}z$ddl}||j|j|jdg7}WntyT||7}Yn0tt|}tt |}t ||dt |j fddt |D}t ||d t}t|||d|d }d }||ksJ|dt|}t |D]\} } |jt| d} t| r@t| sNJn| | ksNJ| } t| dt| d f} |j | d }t| rt|sJn|| ksJqdS) Nnanz+infz-infrZ nodefloatcsi|]\}}||qSr r ).0ivaluerr r 3z6TestGraph.test_special_float_label..Z edgefloatrEagraph [ 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@)floatnumpyrUinf ImportErrorr Z cycle_graphlendict enumerateZset_node_attributeslistrZset_edge_attributesr#rIrJr;rKrLrMr(rstrmathisnan)rZspecial_floatsnpr attrsrOrr6graphZindxrYZ node_valueZedgeZ string_edgeZ edge_valuer rZr test_special_float_label&s>     >     z"TestGraph.test_special_float_labelcCsFtd}d|jdksJtd}d|jks4Jd|jvsBJdS)Nz*graph [ name "x" node [ id 0 label "x" ] ]xrGz!graph [ node [ id 0 label "x" ] ])r rrjrGrr r r test_names   zTestGraph.test_namecCsdD]}dD]}d}|dur0|dtt|7}|durL|dtt|7}|d7}|d7}|d7}t|}t||ksJt||ksJd}|d ur|d 7}|d ur|d 7}|d 7}|r|d 7}|d7}|dt|ks Jq qdS)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/) reintr rboolZ is_directedZ is_multigraphr4r5)rZdirectedZ multigraphgmlr r r r test_graph_typess. zTestGraph.test_graph_typesc CsNddddddddtd d id gg}|td |td t}||_||jd<|jddd|id|jddd|dd tj |t d}tj |t d}||jksJ||d|jksJt|jddddd|idfgksJt|jddddd|dfgksJt}d|jd<tj t |td}|jddksJJdS)NTFl Fx:^VgF"'z"&&&"")DD)r@)rw3iDDz{2.3j, 1 - 2.3j, ()}rrrV)rorg)r]rr/r0Z destringizer)rGrrfrozenset([1, 2, 3]))rHappendrr r2rGrjr3rAr4r5rrrrdrr)rrr rqr r r test_data_typess2  (( zTestGraph.test_data_typescCsXd}t|}dtdd|jks(Jdt|}d}d|d}||ksTJdS) NzTgraph [ name "&"䑄��&unknown;" ]z&"rvz*��&unknown;r/z##1234567890;&#x1234567890abcdefz,graph [ name "&"䑄&z;&unknown;" ])r rrHrGr4r5)rrqr Zalnur6r r r test_escape_unescapes zTestGraph.test_escape_unescapecCstttdtttdttttttttgdttttt<}|t j d| dtt j t j|Wdn1s0Ydd}|dd 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-<||td.dS)/N(rzr@rwrsgraph[]rcSsttjtj|dSN)r<r=r r>r)rqr r r assert_parse_errorsz5TestGraph.test_exceptions..assert_parse_errorz graph [ ]rmz 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 ]uKgraph [edge [ source u'uĐ0' target u'uĐ0' ] node [ id u'uĐ0' label b ] ]csttjfdddS)NcsttjiSr)rdr r5r argskwargsr r 3r\zJTestGraph.test_exceptions..assert_generate_error..)r<r=r r>rr rr assert_generate_error1sz8TestGraph.test_exceptions..assert_generate_errorrrxrr0)r<r= ValueErrorrr frozensetr# TemporaryFiler&codecsBOM_UTF8r;r r>r(rr2rj)rfrrr r r r test_exceptionss  0     zTestGraph.test_exceptionscstj|jddtjgdks&JfddtjD}|gdksNJtj|jddtjgdkstJfddtjD}|gdksJdS)Nrrrcsg|]}j|dqSrrrWnr r r Cr\z.TestGraph.test_label_kwarg..rcsg|]}j|dqSrrrrr r rHr\)r rrrr)rlabelsr rr test_label_kwarg@szTestGraph.test_label_kwargc Cst}dddddddd}|jd i|t\}}zxt||t|}|jd D]:\}}|d ksv|d krt |t ksJq^t |t ks^Jq^Wt |t |nt |t |0dS) Nliril)toosmallsmallZmed1Zmed2Zmed3bigtoobigNoderr)r)r r2r3r#r$rJr(ritemstypereror)r'r*)rr Znumbersr+r,ZG2rNrYr r r test_outofrange_integersKs,      z"TestGraph.test_outofrange_integerscCs*d}t|}|jddddks&JdS)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_multilinejs  zTestGraph.test_multiline)__name__ __module__ __qualname__ classmethodr rr!r.r7r?r<mark parametrizerrBrPrSrTrkrnrrr|r}rrrrr r r r rs* +U   f ] rccst}|V|ddS)Nr)r8r9r;)Z _file_handler r r byte_filesrc@s,eZdZddZddZddZddZd S) TestPropertyListscCslt}|jdgddt}t||Wdn1sB0Y|}|tdkshJdS)Nn1)elementrr@@TF propertiesa7 graph [ node [ id 0 label "n1" properties "element" properties 0 properties 1 properties 2.5 properties 1 properties 0 ] ] r r2r3rrJrKrMrrgrresultr r r 3test_writing_graph_with_multi_element_property_lists* zETestPropertyLists.test_writing_graph_with_multi_element_property_listcCsjt}|jddgdt}t||Wdn1s@0Y|}|tdksfJdS)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_listcCsrt8}|tdd|dt|}Wdn1sD0Y|jddddgdiksnJdS) Na graph [ node [ id 0 label "n1" properties "element" properties 0 properties 1 properties 2.5 ] ] rErTrrr)rrr@rrr&rr:r;r r(rrrrjr r r %test_reading_graph_with_list_propertys  (z7TestPropertyLists.test_reading_graph_with_list_propertycCspt8}|tdd|dt|}Wdn1sD0Y|jdddddgikslJdS) Nz graph [ node [ id 0 label "n1" properties "_networkx_list_start" properties "element" ] ] rErTrrrrrrr r r 4test_reading_graph_with_single_element_list_propertys  (zFTestPropertyLists.test_reading_graph_with_single_element_list_propertyN)rrrrrrrr r r r rsrcollr cCstd}||jdd<t}t|||dt|}|jdd|ksVJt|ddd}tj ||szJ|dtj|t d}tj ||sJdS)Nrwrtestrr@)r1ry) r rFrr8r9rJr;r(Z relabel_nodesutilsZ graphs_equalro)rr rHr r r test_stringize_empty_list_tuples     r)rr8rfr)r#astr contextlibrtextwraprr<Znetworkxr Znetworkx.readwrite.gmlrrrrrrrrr r r r s(   u X