ó ŸÃÒYc@s…dZddlmZmZmZddlZddlZddlZddlZ ddl m Z m Z m Z ddd„ƒYZdS(s Unit tests for edgelists. iÿÿÿÿ(t assert_equalt assert_raisestassert_not_equalN(tassert_edges_equaltassert_nodes_equaltassert_graphs_equalt TestEdgelistcBs¡eZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „Zd„Zd„Zd„ZRS(cCs¦tjddƒ|_ddddddg}|jj|ƒ|jjd ƒtj|jƒ|_tjƒ|_|jj ddddgƒtj |jƒ|_ dS(Ntnamettesttatbtctdtetftgiiiii*(R R (R R (R R (R R (R R(R R(iii(iii(iii(iii*( tnxtGraphtGtadd_edges_fromtadd_nodetDiGraphtDGt MultiGraphtXGtadd_weighted_edges_fromt MultiDiGraphtXDG(tselfR ((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pytsetUpscCsGd}tj|ƒ}tj|dtƒ}t|jƒddgƒdS(Ns&# comment line 1 2 # comment line 2 3 tnodetypeiii(ii(ii(tiotBytesIORt read_edgelisttintRtedges(RtstbytesIOR((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_read_edgelist_1scCs°d}tj|ƒ}tj|dtdtƒ}t|jƒd d gƒtj|ƒ}tj|dtƒ}t|jdt ƒddidd6fddid d6fgƒdS( Ns.# comment line 1 2 2.0 # comment line 2 3 3.0 Rtdataiiig@tweightg@(ii(ii( RR RR!R"tFalseRR#tread_weighted_edgelisttTrue(RR$R%R((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_read_edgelist_2%scCs¶d}tj|ƒ}tj|dtdtƒ}t|jƒd d gƒtj|ƒ}tj|dtdtƒ}t|jdtƒddidd6fddid d6fgƒdS( NsD# comment line 1 2 {'weight':2.0} # comment line 2 3 {'weight':3.0} RR'iiig@R(g@(ii(ii( RR RR!R"R)RR#R+(RR$R%R((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_read_edgelist_35scCsetjƒ}tjƒ}|jddgƒtj||dtƒ|jdƒt|j ƒdƒdS( NiiiR'is1 2 2 3 (ii(ii( RR Rt OrderedGraphRtwrite_edgelistR)tseekRtread(RtfhR((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_write_edgelist_1Es    cCsetjƒ}tjƒ}|jddgƒtj||dtƒ|jdƒt|j ƒdƒdS( NiiiR'is1 2 {} 2 3 {} (ii(ii( RR RR.RR/R+R0RR1(RR2R((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_write_edgelist_2Ms    cCs~tjƒ}tjƒ}|jddddƒ|jddddƒtj||dtƒ|jdƒt|j ƒd ƒdS( NiiR(g@ig@R'is(1 2 {'weight': 2.0} 2 3 {'weight': 3.0} ( RR RR.tadd_edgeR/R+R0RR1(RR2R((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_write_edgelist_3Us   cCstjƒ}tjƒ}|jddddƒ|jddddƒtj||ddgƒ|jdƒt|jƒd ƒdS( NiiR(g@ig@R'is1 2 2.0 2 3 3.0 ( RR RR.R5R/R0RR1(RR2R((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_write_edgelist_4^s   cCstjƒ}yDtdƒtdƒtdƒ}tdƒtdƒtdƒ}WnQtk r£tdƒtdƒtdƒ}tdƒtdƒtdƒ}nX|j|did|6tjƒ\}}tj||ƒtj |ƒ}t ||ƒt j |ƒt j |ƒdS( Ni( i{ii§iiDt Radioheadi(RRtchrt ValueErrortunichrR5ttempfiletmkstempR/R!Rtostclosetunlink(RRtname1tname2tfdtfnametH((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyt test_unicodegs  $  $  cCs tjƒ}yDtdƒtdƒtdƒ}tdƒtdƒtdƒ}WnQtk r£tdƒtdƒtdƒ}tdƒtdƒtdƒ}nX|j|did|6tjƒ\}}tt tj ||d d ƒt j |ƒt j |ƒdS( Ni( i{ii§iiDR8itencodingslatin-1(RRR9R:R;R5R<R=RtUnicodeEncodeErrorR/R>R?R@(RRRARBRCRD((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_latin1_issuews  $  $ cCsútjƒ}y4tdƒ}dtdƒd}tdƒd}Wn5tk rwdtdƒd}tdƒd}nX|j|did|6tjƒ\}}tj||d d ƒtj |d d ƒ}t ||ƒt j |ƒt j |ƒdS( NiÝtBjiötrkiÜtberR8iRGslatin-1(RRR9R:R;R5R<R=R/R!RR>R?R@(RRtblurbRARBRCRDRE((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyt test_latin1‡s     cCs¿|j}tjƒ\}}tj||ƒtj|ƒ}tj|ƒ}t||ƒ|jdƒtt |ƒt |ƒƒt t |j ƒƒt |j ƒƒƒt j |ƒt j|ƒdS(NR(RR<R=RR/R!Rt remove_nodeRtlistRR#R>R?R@(RRRCRDREtH2((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_edgelist_graph˜s   % cCs×|j}tjƒ\}}tj||ƒtj|dtjƒƒ}tj|dtjƒƒ}t||ƒ|jdƒt t |ƒt |ƒƒt t |j ƒƒt |j ƒƒƒt j|ƒt j|ƒdS(Nt create_usingR(RR<R=RR/R!RRRORRPRR#R>R?R@(RRRCRDRERQ((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_edgelist_digraph¥s   % cCsÁtj|jƒ}tjƒ\}}tj||ƒtj|dtƒ}|jt tj |ƒƒƒt t |ƒt |ƒƒt t |j ƒƒt |j ƒƒƒtj|ƒtj|ƒdS(NR(Rtconvert_node_labels_to_integersRR<R=R/R!R"tremove_nodes_fromRPtisolatesRRR#R>R?R@(RRRCRDRE((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_edgelist_integers²s% cCs×|j}tjƒ\}}tj||ƒtj|dtjƒƒ}|jdƒtj|dtjƒƒ}t||ƒt t |ƒt |ƒƒt t |j ƒƒt |j ƒƒƒt j|ƒt j|ƒdS(NRSR(RR<R=RR/R!RRORRRPRR#R>R?R@(RRRCRDRERQ((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyRT¾s   % cCsÖ|j}tjƒ\}}tj||ƒtj|dtdtjƒƒ}tj|dtdtjƒƒ}t||ƒt t |ƒt |ƒƒt t |j ƒƒt |j ƒƒƒt j|ƒt j|ƒdS(NRRS(RR<R=RR/R!R"RRRRPRR#R>R?R@(RRRCRDRERQ((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_edgelist_multigraphËs !! % cCsÖ|j}tjƒ\}}tj||ƒtj|dtdtjƒƒ}tj|dtdtjƒƒ}t||ƒt t |ƒt |ƒƒt t |j ƒƒt |j ƒƒƒt j|ƒt j|ƒdS(NRRS(RR<R=RR/R!R"RRRRPRR#R>R?R@(RRRCRDRERQ((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyttest_edgelist_multidigraph×s !! % (t__name__t __module__RR&R,R-R3R4R6R7RFRIRNRRRTRXRYRZ(((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyRs"        ((t__doc__t nose.toolsRRRRR<R>tnetworkxRtnetworkx.testingRRRR(((s|/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/readwrite/tests/test_edgelist.pyts