ó ,µ[c@sÐdZy<yddlmZWn!ek r@ddlmZnXWn!ek reddlmZnXddlZddlZddlmZm Z m Z ddl Z ddl mZdefd„ƒYZdS(s'Unit tests for pydot drawing functions.iÿÿÿÿ(tStringION(t assert_equaltassert_is_instancet assert_true(tassert_graphs_equalt TestPydotcBs;eZed„ƒZd„Zd„Zd„Zd„ZRS(cCs/tjjtjtƒatdk s+t‚dS(sœ Fixture defining the `pydot` global to be the `pydot` module if both importable and of sufficient version _or_ skipping this test. N( tnxtnx_pydott setup_moduletsystmodulest__name__tpydottNonetAssertionError(tcls((s@lib/python2.7/site-packages/networkx/drawing/tests/test_pydot.pyt setupClassscCsðd|jd<|jddddgƒ|jdƒtjj|d|ƒ}t|tƒtjj|ƒ}|j tjj |ƒƒ}t ||ƒt j ƒ}|j|ƒtjd |d d ƒ}tt|ƒd ƒ|d }tg|jƒD]} | jƒ^qðƒ} tg|jƒD]} | jƒ^qƒ} t| | ƒtg|jƒD]} | jƒ| jƒf^qSƒ} tg|jƒD]} | jƒ| jƒf^qŠƒ}t| |ƒtjj|ƒ}|j |ƒ}t ||ƒdS(s© Validate :mod:`pydot`-based usage of the passed NetworkX graph with the passed basename of an external GraphViz command (e.g., `dot`, `neato`). tGtnametAtBtCtDtEtprogtpathtencodingsutf-8iiN(RR(RR(RR(RR(tgraphtadd_edges_fromtadd_nodeRRt pydot_layoutRtdicttto_pydott __class__t from_pydotRttempfiletmktempt write_rawR tgraph_from_dot_fileRtlentsortedt get_node_listtget_namet get_edge_listt get_sourcetget_destinationtread_dot(tselfRRt graph_layouttPtG2tfnametPin_listtPintptn1tn2tete1te2tHin((s@lib/python2.7/site-packages/networkx/drawing/tests/test_pydot.pyt pydot_checkss0       ++ 44 cCs|jtjƒddƒdS(NRtneato(R=RtGraph(R/((s@lib/python2.7/site-packages/networkx/drawing/tests/test_pydot.pyttest_undirected^scCs|jtjƒddƒdS(NRtdot(R=RtDiGraph(R/((s@lib/python2.7/site-packages/networkx/drawing/tests/test_pydot.pyt test_directedascCs{tjƒ}d|jd<|jddddƒtƒ}tjj||ƒ|jdƒtjj|ƒ}t ||ƒdS(NRRt1t2tkeyt0i( Rt MultiGraphRtadd_edgeRRt write_dottseekR.R(R/RtfhtH((s@lib/python2.7/site-packages/networkx/drawing/tests/test_pydot.pyttest_read_writeds    (R t __module__t classmethodRR=R@RCRN(((s@lib/python2.7/site-packages/networkx/drawing/tests/test_pydot.pyRs  C  (t__doc__t cStringIORt ImportErrortioR R#t nose.toolsRRRtnetworkxRtnetworkx.testingRtobjectR(((s@lib/python2.7/site-packages/networkx/drawing/tests/test_pydot.pyts