U mdVQ@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 Z/home/sam/Atlas/atlas_env/lib/python3.8/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|dddgks*tt|dddgksDtt|jd d ddd d d ddfddddifddddifgkstdS)NlabelrNode 1Node 2Node 3)rr)rr)rrTdatablue)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@rr0r1zgraph [ node [ id 0 label "(0,1)" ] node [ id 1 label "(1,0)" ] edge [ source 0 target 1 ] ])r r2add_edger4r5rr)rr!rr6r 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!attrfobjrr6r r r test_quotess    zTestGraph.test_quotescCsbdtd}t}||t}t|||d| d}d}||ks^t dS)NnoderCrrDz6graph [ node [ id 0 label "node©" ] ]) rGr r2r3r$rHrIr;rJrKrLrrrPr!rNrr6r r r test_unicode_nodes    zTestGraph.test_unicode_nodecCsZd}t}||t}t|||d| d}d}||ksVt dS)Ng?rrDz/graph [ node [ id 0 label "1.0" ] ]) r r2r3r$rHrIr;rJrKrLrrQr 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 edgefloatrDagraph [ 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@)floatnumpyrTinf ImportErrorr Z cycle_graphlendict enumerateZset_node_attributeslistrZset_edge_attributesr$rHrIr;rJrKrLrr)rstrmathisnan)rZspecial_floatsnpr!attrsrNrr6graphZindxrXZ node_valueedgeZ string_edgeZ edge_valuer rYr 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" ] ]xrFz!graph [ node [ id 0 label "x" ] ])r rrhrrFr r 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 ] ]r0) rcintr rboolZ is_directedrZ 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}||jkst||d|jkstt|jddddd|idfgkstt|jddddd|dfgkstt}d|jd<tj t |td}|jddksJtdS)NTFl Fx:^VgF"'z"&&&"")DD)r@)rv3iDDz{2.3j, 1 - 2.3j, ()}rrrU)rnrg)r[rr0r1Z destringizer)rFrrfrozenset([1, 2, 3]))rGappendrr r2rFrhr3rAr4r5rrrrrbrr)rrr!rpr r r test_data_typess2  (( zTestGraph.test_data_typescCsXd}t|}dtdd|jks(tdt|}d}d|d}||ksTtdS) NzTgraph [ name "&"䑄��&unknown;" ]z&"ruz*��&unknown;r0z##1234567890;&#x1234567890abcdefz,graph [ name "&"䑄&z;&unknown;" ])r rrGrFrr4r5)rrpr!Zalnur6r 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(ryr@rvrsgraph[]rcSsttjtj|dSN)r<r=r r>r)rpr r r assert_parse_errorsz5TestGraph.test_exceptions..assert_parse_errorz graph [ ]rlz 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~)rbr r5r argskwargsr r 2zJTestGraph.test_exceptions..assert_generate_error..)r<r=r r>rr rr assert_generate_error0s z8TestGraph.test_exceptions..assert_generate_errorrwrr1)r<r= ValueErrorrr frozensetr$ TemporaryFiler'codecsBOM_UTF8r;r r>r)rr2rh)rfrrr!r r r test_exceptionss       zTestGraph.test_exceptionscstj|jddtjdddgks(tfddtjD}|dd d gksRttj|jddtjdddgksztfd dtjD}|dd d gkstdS) Nrrr@rvrcsg|]}j|dqSrrrVnr!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)toosmallZsmallZmed1Zmed2Zmed3bigtoobigNoderr)r)r r2r3r$r%r*r(r+rIr)ritemstypercrrn)rr!numbersr,r-ZG2rMrXr r r test_outofrange_integersJs(     z"TestGraph.test_outofrange_integersN)__name__ __module__ __qualname__ classmethodr rr"r/r7r?rBrOrRrSrjrmrqr{r|rrrr r r r rs& +U  f ] rccst}|V|ddS)Nr)r8r9r;)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 r2r3rrIrJrLrrrgrresultr 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 ] ] rDrTrrrrr@r rr'rr:r;r r)rrrrrhr 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" ] ] rDrTrrrrrrr 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)Nrvrtestrr@)r1rx) r rErr8r9rIr;r)rZ relabel_nodesutilsZ graphs_equalrn)rr!rHr r r test_stringize_empty_list_tuples     r)rr8rdr*r$astr contextlibrtextwraprr<Znetworkxr Znetworkx.readwrite.gmlrrrrrmarkZ parametrizerr r r r s(   ^ X