U :v fVQ@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 d dZ ddZ ddZ ddZ ddZddZddZddZddZdd Zd!d"Zd#d$Zd%S)& 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 E/tmp/pip-unpacked-wheel-_lngutwb/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|dddgks*tt|dddgksDtt|jd d ddd d d ddfddddifddddifgkstdS)NlabelrNode 1Node 2Node 3)rr)rr)rrTdataZblue)lineZ thicknesszEdge from node 1 to node 2)colorrzEdge from node 2 to node 3zEdge from node 3 to node 1)r rrsortednodesAssertionErroredgesrGr r r test_parse_gmls"    zTestGraph.test_parse_gmlcCst\}}t|d}||j|tj|dd}tj|jdd}t |j ddt |j ddkslt t |j ddt |j ddkst t |t |dS)NwrrTr)tempfilemkstempopenwritercloser read_gmlrrrrrosunlink)rfdfnamefhZGinr r r r test_read_gmls   $$ zTestGraph.test_read_gmlcCs:d}t}|ddtj|td}||ks6tdS)Nz0graph [ node [ id 0 label "1203" ] ]i Z stringizer)r Graphadd_nodejoin generate_gmlrr)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_duplicatecCs<t}|dddtj|td}d}||ks8tdS)N)r)r?rr/r0zgraph [ node [ id 0 label "(0,1)" ] node [ id 1 label "(1,0)" ] edge [ source 0 target 1 ] ])r r1add_edger3r4rr)rr rr5r r r test_tuplelabelss  zTestGraph.test_tuplelabelscCsntd}d|_dtd}||jdd<t}t|||d| d}d}||ksjt dS) Nr?z path_graph(1)z*This is "quoted" and this is a copyright: rZdemoasciizgraph [ name "path_graph(1)" node [ id 0 label "0" demo "This is "quoted" and this is a copyright: ©" ] ]) r path_graphnamechrrr#NamedTemporaryFile write_gmlr:readstripdecoder)rr attrfobjrr5r r r test_quotess    zTestGraph.test_quotescCsbdtd}t}||t}t|||d| d}d}||ks^t dS)NnoderBrrCz6graph [ node [ id 0 label "node©" ] ]) rFr r1r2r#rGrHr:rIrJrKrrrOr rMrr5r r r test_unicode_nodes    zTestGraph.test_unicode_nodecCsZd}t}||t}t|||d| d}d}||ksVt dS)Ng?rrCz/graph [ node [ id 0 label "1.0" ] ]) r r1r2r#rGrHr:rIrJrKrrPr r r test_float_labels   zTestGraph.test_float_labelcstdtdtdg}z$ddl}||j|j|jdg7}Wntk rV||7}YnXtt|}tt |}t ||dt |j fddt |D}t ||d t}t|||d|d }d }||kst|dt|}t |D]\} } |jt| d} t| rBt| sPtn| | ksPt| } t| dt| d f} |j | d }t| rt|stn|| ks tq dS) Nnanz+infz-infrZ nodefloatcsi|]\}}||qSr r ).0ivaluerr r 2sz6TestGraph.test_special_float_label..Z edgefloatrCagraph [ 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?)floatZnumpyrSinf ImportErrorr Z cycle_graphlendict enumerateZset_node_attributeslistrZset_edge_attributesr#rGrHr:rIrJrKrr(rstrmathisnan)rZspecial_floatsnpr attrsrMrr5graphZindxrWZ node_valueZedgeZ string_edgeZ edge_valuer rXr test_special_float_label%s>    >     z"TestGraph.test_special_float_labelcCsFtd}d|jdksttd}d|jks4td|jksBtdS)Nz*graph [ name "x" node [ id 0 label "x" ] ]xrEz!graph [ node [ id 0 label "x" ] ])r rrfrrErr r r test_names   zTestGraph.test_namecCsdD]}dD]}d}|dk r0|dtt|7}|dk rL|dtt|7}|d7}|d7}|d7}t|}t||kstt||kstd}|d kr|d 7}|d kr|d 7}|d 7}|r|d 7}|d7}|dt |ks tq 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/) raintr rboolZ is_directedrZ is_multigraphr3r4)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}||jkst||d|jkstt|jddddd|idfgkstt|jddddd|dfgkstt}d|jd<tj t |td}|jddksJtdS)NTFl Fx:^VgF"'z"&&&"")DD)r?)rs3iDDz{2.3j, 1 - 2.3j, ()}rrrT)rkrg)rZrr/r0Z destringizer)rErrfrozenset([1, 2, 3]))rFappendrr r1rErfr2r@r3r4rrrrr`rr)rrr rmr r r test_data_typess2  (( zTestGraph.test_data_typescCsXd}t|}dtdd|jks(tdt|}d}d|d}||ksTtdS) NzTgraph [ name "&"䑄��&unknown;" ]z&"rrz*��&unknown;r/z##1234567890;&#x1234567890abcdefz,graph [ name "&"䑄&z;&unknown;" ])r rrFrErr3r4)rrmr Zalnur5r r r test_escape_unescapes zTestGraph.test_escape_unescapec Cstttdtttdttttttttdddgttttt2}|t j d| dtt j t j|W5QRXdd }|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 }tdddg|jd.<||td/dS)0N(rvr?rsrsgraph[]rcSsttjtj|dSN)r;r<r r=r)rmr r r assert_parse_errorsz5TestGraph.test_exceptions..assert_parse_errorz graph [ ]riz 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)NcsttjSr{)r`r r4r argskwargsr r 2zJTestGraph.test_exceptions..assert_generate_error..)r;r<r r=rr rr assert_generate_error0s z8TestGraph.test_exceptions..assert_generate_errorrtrr0)r;r< ValueErrorrr frozensetr# TemporaryFiler&codecsBOM_UTF8r:r r=r(rr1rf)rfr|rr r r r test_exceptionss       zTestGraph.test_exceptionscstj|jddtjdddgks(tfddtjD}|dd d gksRttj|jddtjdddgksztfd dtjD}|dd d gkstdS) Nrrr?rsrcsg|]}j|dqSrrrUnr r r Bsz.TestGraph.test_label_kwarg..rrrcsg|]}j|dqSrrrrr r rGs)r rrrrr)rlabelsr rr test_label_kwarg?szTestGraph.test_label_kwargc Cst}dddddddd}|jd |t\}}zdt||t |}|j d  D]:\}}|d ksr|d krt |t kstqZt |tksZtqZW5t|t|XdS) Nliril)toosmallsmallZmed1Zmed2Zmed3bigtoobigNoderr)r)r r1r2r#r$r)r'r*rHr(ritemstyperarrk)rr Znumbersr+r,ZG2rLrWr r r test_outofrange_integersJs(     z"TestGraph.test_outofrange_integersN)__name__ __module__ __qualname__ classmethodr rr!r.r6r>rArNrQrRrgrjrnrxryrrrr r r r rs& +U  f ] rccst}|V|ddS)Nr)r7r8r:)Z _file_handler r r byte_filejsrc@s,eZdZddZddZddZddZd S) TestPropertyListsc Cs`t}|jdddddddgdt}t||W5QRX|}|td ks\tdS) Nn1elementrr?@TF propertiesa7 graph [ node [ id 0 label "n1" properties "element" properties 0 properties 1 properties 2.5 properties 1 properties 0 ] ] r r1r2rrHrIrKrrrgrresultr r r 3test_writing_graph_with_multi_element_property_listrs zETestPropertyLists.test_writing_graph_with_multi_element_property_listc CsVt}|jddgdt}t||W5QRX|}|tdksRtdS)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_listc Csbt.}|tdd|dt|}W5QRX|jddddddd d giks^tdS) Na graph [ node [ id 0 label "n1" properties "element" properties 0 properties 1 properties 2.5 ] ] rCrTrrrrr?r rr&rr9r:r r(rrrrrfr r r %test_reading_graph_with_list_propertys  z7TestPropertyLists.test_reading_graph_with_list_propertyc Cs\t.}|tdd|dt|}W5QRX|jdddddgiksXtdS) 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 rqsrcollr cCstd}||jdd<t}t|||dt|}|jdd|ksVtt |ddd}tj ||szt|dtj|t d}tj ||stdS)Nrsrtestr~r?)r~1ru) r rDrr7r8rHr:r(rZ relabel_nodesutilsZ graphs_equalrk)rr rHr r r test_stringize_empty_list_tuples     r)rr7rbr)r#astr contextlibrtextwraprr;Znetworkxr Znetworkx.readwrite.gmlrrrrrmarkZ parametrizerr r r r s(   ^ X