ó
,µ[c @ s° d d l Td d l m Z d d l Z d d l Td d l Z d d l Z d d l Z d e f d „ ƒ YZ
d e
f d „ ƒ YZ d e
f d „ ƒ YZ d
e f d „ ƒ YZ
d S( iÿÿÿÿ( t *( t SkipTestNt BaseGraphMLc B s e Z d „ Z RS( c C s( d | _ t j ƒ | _ | j j d ƒ | j j d d d d ƒ| j j d' d( d) d* d+ d, d- d. d/ d0 g
ƒ t j | j j d ƒ ƒ | _
d | _ t j d d ƒ | _ i d d 6| j j
d <| j j d d d ƒ| j j d d d ƒ| j j d d d ƒ| j j d
ƒ | j j d d d ƒ| j j d d d d d d ƒ| j j d d d d d d ƒ| j j d d d d d d ƒ| j j d d d d ƒ| j j d d
d d ƒ| j j d d d d! ƒ| j j d d
d d" d d# ƒt j | j j d ƒ ƒ | _ d$ | _ t j ƒ | _ | j j d d d% ƒ| j j d d d ƒ| j j d d d d% ƒ| j j d d d d ƒt j | j j d ƒ ƒ } | | _ d& | _ t j ƒ | _ | j j d ƒ | j j d d d d ƒ| j j d1 d2 g ƒ t j | j j d ƒ ƒ } | | _ d S(3 Ns
t n10t n0t n2t idt foot n1t n3t n5t n4t n6t n7t n8t n9s UTF-8s
yellow
green
blue
red
turquoise
1.0
1.0
2.0
1.1
t Gt yellowt colort node_defaultt greent bluet redt turquoiset e0t weightg ð?t e1t e2g @t e3t e4t e5t e6gš™™™™™ñ?sC
1
2.0
1
k
1.0
i s
( R R ( R R ( R R
( R R ( R R ( R R
( R
R
( R R ( R R
( R R ( R R ( R R ( t simple_directed_datat nxt DiGrapht simple_directed_grapht add_nodet add_edget add_edges_fromt iot BytesIOt encodet simple_directed_fht attribute_datat attribute_grapht grapht attribute_fht attribute_numeric_type_datat attribute_numeric_type_grapht attribute_numeric_type_fht simple_undirected_datat Grapht simple_undirected_grapht simple_undirected_fh( t selft fh( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt setUp s^
+
( t __name__t
__module__R8 ( ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyR s t TestReadGraphMLc B s} e Z d „ Z d „ Z d „ Z d „ Z d „ Z d „ Z d „ Z d „ Z d „ Z
d „ Z d
„ Z d „ Z
d „ Z RS(
c C s7 | j } t j | j ƒ } t t | j ƒ ƒ t | j ƒ ƒ ƒ t t | j ƒ ƒ t | j ƒ ƒ ƒ t t | j d t ƒ ƒ t | j d t ƒ ƒ ƒ | j j d ƒ t j
| j ƒ } t t | j ƒ ƒ t | j ƒ ƒ ƒ t t | j ƒ ƒ t | j ƒ ƒ ƒ t t | j d t ƒ ƒ t | j d t ƒ ƒ ƒ d S( Nt datai ( R# R! t read_graphmlR* t assert_equalt sortedt nodest edgest Truet seekt
parse_graphmlR ( R6 R t Ht I( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt! test_read_simple_directed_graphml´ s %%%%c C s¥ | j } t j | j ƒ } t | j ƒ | j ƒ ƒ t | j ƒ | j ƒ ƒ | j j d ƒ t j | j
ƒ } t | j ƒ | j ƒ ƒ t | j ƒ | j ƒ ƒ d S( Ni ( R4 R! R= R5 t assert_nodes_equalR@ t assert_edges_equalRA RC RD R2 ( R6 R RE RF ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt# test_read_simple_undirected_graphmlà s c C sQ | j } t j | j ƒ } t | j t ƒ t | j d t ƒ ƒ ƒ t | j d t ƒ ƒ } t | j d t ƒ ƒ } x* t | | ƒ D] \ } } t
| | ƒ qƒ W| j j d ƒ t j | j
ƒ } t
t | j t ƒ ƒ t | j d t ƒ ƒ ƒ t | j d t ƒ ƒ } t | j d t ƒ ƒ } x* t | | ƒ D] \ } } t
| | ƒ q0Wd S( NR< i ( R, R! R= R. RH R@ RB R? RA t zipR> RC RD R+ ( R6 R RE t get het at bRF ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_read_attribute_graphmlÎ s ( . c C sN d } t j | j d ƒ ƒ } t t j t j | ƒ t t j t j | ƒ d S( NsÛ
s UTF-8( R' R( R) t
assert_raisesR! t
NetworkXErrorR= RD ( R6 t sR7 ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_directed_edge_in_undirectedß s
c C sN d } t j | j d ƒ ƒ } t t j t j | ƒ t t j t j | ƒ d S( Nsó
s UTF-8( R' R( R) RQ R! RR R= RD ( R6 RS R7 ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_undirected_edge_in_directedñ s
c C sN d } t j | j d ƒ ƒ } t t j t j | ƒ t t j t j | ƒ d S( Nsô
yellow
green
blue
1.0
s UTF-8( R' R( R) RQ R! RR R= RD ( R6 RS R7 ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_key_raise s c C sN d } t j | j d ƒ ƒ } t t j t j | ƒ t t j t j | ƒ d S( Ns5
yellow
green
blue
s UTF-8( R' R( R) RQ R! RR R= RD ( R6 RS R7 ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_hyperedge_raise s c C s— d } t j | j d ƒ ƒ } t j | ƒ } d d
g } t t | j d t ƒ ƒ | ƒ | j d ƒ t j
| ƒ } t t | j d t ƒ ƒ | ƒ d S( Nsß
s UTF-8R R R R t keysi ( R R R ( R R R ( R' R( R) R! R= R> R? RA RB RC RD ( R6 RS R7 R t expectedRE ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_multigraph_keys= s
c C sO t j ƒ } | j d ƒ | j d ƒ | j d d d t d d ƒ f d d t d d ƒ f d d t d d d d ƒ f d d d t d d
ƒ f d d d t d d ƒ f d d d
t d d d d ƒ f g ƒ t j ƒ } t j | | ƒ | j d ƒ t j | d t
ƒ} t | j d t
d t
ƒ | j d t
d t
ƒ ƒ t | j | j ƒ d S( sf
Test that data and keys of edges are preserved on consequent
write and reads
i i t keyt data_key1R t data_id2t data_key3t data_id3ig t data_key4ih t data_id5ii t data_key6t data_id7i t node_typeR< RX N( i i ( R! t
MultiGraphR$ R& t dictR' R( t
write_graphmlRC R= t intRI RA RB R> t _adj( R6 R R7 RE ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_preserve_multi_edge_dataT s$
%
.c C s
d } t j | j d ƒ ƒ } t j | ƒ } t t | j ƒ ƒ d
g ƒ t | d d d d ƒ t | j d d d ƒ t | j d d d ƒ t j | ƒ } t t | j ƒ ƒ d g ƒ t | d d d d ƒ t | j d d d ƒ t | j d d d ƒ d S( Ns†
1
2
s UTF-8R R R R t labelt 1t 2( R R ( R R (
R' R( R) R! R= R> t listRA R@ RD ( R6 R< R7 R RE ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_yfiles_extensionq s C c C sç d } t j | j d ƒ ƒ } t j | ƒ } t j | ƒ } x¤ | | g D]– } t | j d d t ƒ t | j d d t ƒ t | j d d t ƒ t | j d d t ƒ t | j d d t ƒ t | j d d t ƒ qI Wd S(
NsU
false
true
false
FaLsE
True
0
1
s UTF-8R t testR R R R
R (
R' R( R) R! R= RD R> R@ RB t False( R6 RS R7 R RE R- ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_bool s c C sÖ d } d } d } xy | | f D]k } t j | j d ƒ ƒ } t j | ƒ } t j | ƒ } x, | | g D] } t | j d d t ƒ qh Wq Wt j | j d ƒ ƒ } t t j
t j | ƒ t t j
t j | ƒ d S( Nsþ
false
true
s
false
true
s%
false
true
s UTF-8R Rp ( R' R( R) R! R= RD R> R@ RB RQ RR ( R6 t goodt badt uglyRS R7 R RE R- ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_graphml_header_lineí s c C s‘ d } t j | j d ƒ ƒ } t j | ƒ } g | j d t ƒ D] \ } } | ^ q@ } t t | ƒ d ƒ x | D] } t | d d ƒ qr Wd S( Ns‡I
2
Group 3
Folder 3
Group 1
Folder 1
1
3
Group 2
Folder 2
5
6
9
s UTF-8R< i t CustomPropertyt (
R' R( R) R! R= R@ RB R> t lent assert_not_equal( R6 R< R7 R t _t xt node_data( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_read_attributes_with_groups s ÿ +
( R9 R: RG RJ RP RT RU RV RW RZ Rj Ro Rr Rv R~ ( ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyR; ³ s Q + 3t TestWriteGraphMLc B s‰ e Z e e j ƒ Z e d „ ƒ Z d „ Z d „ Z d „ Z
d „ Z d „ Z d „ Z
d „ Z d „ Z d „ Z d
„ Z d „ Z RS( c C s4 y d d l } Wn t k
r/ t d ƒ ‚ n Xd S( Niÿÿÿÿs lxml.etree not available.( t
lxml.etreet ImportErrorR ( t clst lxml( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt
setupClassH s
c C sN y# d d l } t t j t j ƒ Wn$ t k
rI t t j t j ƒ n Xd S( Niÿÿÿÿ( R€ R> R! Rg t write_graphml_lxmlR t write_graphml_xml( R6 Rƒ ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_write_interfaceO s
c C sÝ | j } d | j d R? R@ RA RB R* ( R6 R R7 RE ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt' test_write_read_simple_directed_graphmlV s
%%1c C sX d d l m } | j } t j ƒ } | j | | d t ƒ| j d ƒ t j | ƒ } | j d ƒ t
| j ƒ | j ƒ ƒ t | j
ƒ | j
ƒ ƒ t | j
d t ƒ | j
d t ƒ ƒ | j j d ƒ | | ƒ } | j ƒ j ƒ } t t | ƒ d ƒ g | d D] } | j ƒ ^ q} t t | ƒ d ƒ t d | d ƒ t d | d
ƒ d S(
Niÿÿÿÿ( t parset infer_numeric_typesi R< i i s attr.typet doublei ( s attr.typeRŽ ( s attr.typeRŽ ( t xml.etree.ElementTreeRŒ R0 R' R( RŠ RB RC R! R= RH R@ RI RA R1 t getroott getchildrenR> Ry t itemst assert_in( R6 RŒ R R7 RE t xmlt childrent childRX ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt. test_write_read_attribute_numeric_type_graphmlb s$
% #c C s¶ t j ƒ } | j d d g ƒ t j ƒ \ } } | j | | ƒ t j | ƒ } t | j ƒ ƒ t | j
d t ƒ | j
d t ƒ ƒ t | j
| j
ƒ t j | ƒ t j | ƒ d S( så Writing keys as edge id attributes means keys become strings.
The original keys are stored as data, so read them back in
if `make_str(key) == edge_id`
This allows the adjacency to remain the same.
RN RO i i RX N( RN RO i ( RN RO i ( R! Re R& t tempfilet mkstempRŠ R= t assert_truet
is_multigraphRI RA RB R> Ri t ost closet unlink( R6 R t fdt fnameRE ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_more_multigraph_keys| s %
c C s t j d d ƒ } | j d d d d d ƒt j | d d d d g ƒ | j d d d
d ƒi d d 6| j d ( R6 R R7 RE ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_default_attribute s
c C s~ t j ƒ } | j d d g ƒ t j ƒ \ } } | j | | ƒ t j | ƒ } t | j ƒ ƒ t j
| ƒ t j | ƒ d S( NRN RO i t ci ( RN RO i ( RO R§ i ( R! Re R& R˜ R™ RŠ R= t assert_falseR› Rœ R Rž ( R6 R RŸ R RE ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_multigraph_to_graphœ s
c C s´ y d d l } Wn d SX| j d ƒ } t j d d i | d 6f g ƒ } t j ƒ \ } } | j | | ƒ t j | d t ƒ} t | j
| j
ƒ t j | ƒ t j
| ƒ d S( Niÿÿÿÿg333333 @i i R Rd ( t numpyt floatR! R3 R˜ R™ RŠ R= Rh R> Ri Rœ R Rž ( R6 t npt wtR RŸ R RE ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_numpy_float§ s "
c C s y d d l } Wn d SX| j d ƒ } t j d d i | d 6f g ƒ } t j ƒ \ } } | j | | ƒ t j | d t ƒ} t | j
| j
ƒ | d d d } | d d d } t | | d d ƒt t | ƒ | j ƒ t t | ƒ t
ƒ t j | ƒ t j | ƒ d S( Niÿÿÿÿg333333 @i i R Rd t placesi ( Rª t float64R! R3 R˜ R™ RŠ R= Rh R> RA t assert_almost_equalt typeR« Rœ R Rž ( R6 R¬ R R RŸ R RE t wtGt wtH( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_numpy_float64µ s" "
c C s y d d l } Wn d SX| j d ƒ } t j d d i | d 6f g ƒ } t j ƒ \ } } | j | | ƒ t j | d t ƒ} t | j
| j
ƒ | d d d } | d d d } t | | d d ƒt t | ƒ | j ƒ t t | ƒ t
ƒ t j | ƒ t j | ƒ d S( Niÿÿÿÿg333333 @i i R Rd R¯ i ( Rª t float32R! R3 R˜ R™ RŠ R= Rh R> RA R± R² R« Rœ R Rž ( R6 R¬ R R RŸ R RE R³ R´ ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_numpy_float32È s" "
c C s. t j ƒ } yJ t d ƒ t d ƒ t d ƒ } t d ƒ t d ƒ t d ƒ } t } WnW t k
r¯ t d ƒ t d ƒ t d ƒ } t d ƒ t d ƒ t d ƒ } t } n X| j | d d | ƒt j ƒ \ } } | j
| | ƒ t j | d | ƒ} t | j
| j
ƒ t j | ƒ t j | ƒ d S(
Ni( i{ i i§ i iD t RadioheadR Rd ( R! R3 t chrt strt
ValueErrort unichrt unicodeR% R˜ R™ RŠ R= R> Ri Rœ R Rž ( R6 R t name1t name2Rd RŸ R RE ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_unicode_attributesÛ s
c C sÑ d d l } t d d ƒ } y t d ƒ | j | ƒ } Wn& t k
r` t | j | ƒ ƒ } n Xt j ƒ } | | j d ( R6 RÁ RN t saR R7 RE ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt test_unicode_escapeí s
( R9 R: t staticmethodR! R… RŠ t classmethodR„ R‡ R‹ R— R¡ R¦ R© R® Rµ R· RÀ RÄ ( ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyR E s t TestXMLGraphMLc B s& e Z e e j ƒ Z e d „ ƒ Z RS( c C s4 y d d l } Wn t k
r/ t d ƒ ‚ n Xd S( Niÿÿÿÿs$ xml.etree.ElementTree not available.( R R R ( R‚ R” ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyR„ s
( R9 R: RÅ R! R† RŠ RÆ R„ ( ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyRÇ ÿ s ( t
nose.toolst noseR t networkxR! t networkx.testing.utilsR' R˜ Rœ t objectR R; R RÇ ( ( ( sD lib/python2.7/site-packages/networkx/readwrite/tests/test_graphml.pyt s
¨ÿ ÿ ”º