U
:v
fK @ s2 d dl Z d dlZd dlZd dlZG dd dZdS ) Nc @ s e Zd Zedd Zdd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dd Z
dd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-S ).TestGEXFc C s d| _ t | _| jjddd | jjddd | jjdddd t| j d| _ d | _
t | _ d
d i| j j d < | j jdd
dddd | j jdddddd | j jddddd d | j jddddd d | j jddddd | j jdddd | j jdddd | j jdddd | j jddd d t| j
d| _
d | _t | _| jjddd | jjddd | jjdddd t| jd| _d S ) Nar
0Hello)label1ZWorld)idUTF-8a]
Gephi.org
A Web network
true
frogT node_defaultZGephizhttps://gephi.org F)r urlZindegreer ZWebatlaszhttp://webatlas.fr 2ZRTGIzhttp://rtgi.fr3Z BarabasiLabzhttp://barabasilab.comfoor r 4at
)Zsimple_directed_datanxZDiGraphsimple_directed_graphadd_nodeadd_edgeioBytesIOencodesimple_directed_fhZattribute_dataattribute_graphgraph attribute_fhZsimple_undirected_dataGraphsimple_undirected_graphsimple_undirected_fh)cls r" F/tmp/pip-unpacked-wheel-_lngutwb/networkx/readwrite/tests/test_gexf.py setup_class
sf
8
zTestGEXF.setup_classc C s~ | j }t| j}t| t| ks.tt| t| ksJtt|jddt|jddksnt| jd d S NTdatar ) r r read_gexfr sortednodesAssertionErroredgesseekselfGHr" r" r# !test_read_simple_directed_graphml s $z*TestGEXF.test_read_simple_directed_graphmlc C s | j }t }t|| |d t|}t| t| ksJt t|
t|
ksft t|j
ddt|j
ddkst | j d d S )Nr Tr& ) r r r r
write_gexfr- r( r) r* r+ r, r r/ r0 fhr1 r" r" r# 'test_write_read_simple_directed_graphml s
$z0TestGEXF.test_write_read_simple_directed_graphmlc C sn | j }t| j}t| t| ks.ttdd | D tdd | D ks^t| jd d S )Nc s s | ]}t |V qd S Nr) .0er" r" r# s z?TestGEXF.test_read_simple_undirected_graphml..r ) r r r( r r) r* r+ r, r- r. r" r" r# #test_read_simple_undirected_graphml s z,TestGEXF.test_read_simple_undirected_graphmlc C s | j }t| j}t|dt|jddks4tt|jdd}t|jdd}t||D ]\}}||ks^tq^| j d d S r% )
r r r( r r) r* r+ r, zipr- )r/ r0 r1 geheabr" r" r# test_read_attribute_graphml s "z$TestGEXF.test_read_attribute_graphmlc C s* d}t |d}ttjtj| d S )Na
r r r r pytestZraisesr Z
NetworkXErrorr( r/ sr5 r" r" r# test_directed_edge_in_undirected s
z)TestGEXF.test_directed_edge_in_undirectedc C s* d}t |d}ttjtj| d S )Na
r rD rF r" r" r# test_undirected_edge_in_directed s
z)TestGEXF.test_undirected_edge_in_directedc C s* d}t |d}ttjtj| d S )Na
r rD rF r" r" r# test_key_raises s zTestGEXF.test_key_raisesc C s> d}t |d}tj|dd}t| ddgks:td S )Nay
r T)Zrelabelr Word)r r r r r( r) r* r+ )r/ rG r5 r0 r" r" r# test_relabel s
zTestGEXF.test_relabelc C s t }|jdddd t |ddddg |jdddd d d
i|jd < d d
i|jd< t }t || | d t j
|t d}t |
t |
kstt dd | D t dd | D kst|jd= |j|jkstd S )Nr r greenr colorr r
)r rO Zyellowr
r Z edge_defaultZ node_typec s s | ]}t |V qd S r7 r8 r9 r" r" r# r< s z2TestGEXF.test_default_attribute..mode)r r r Zadd_pathr r r r r3 r- r( intr) r* r+ r, r4 r" r" r# test_default_attribute s
z TestGEXF.test_default_attributec C sp t }|jdddd t }t || |d t j|td}t |dgksVt
|j d d dkslt
d S ) Nr rQ M r r rR r Z77) r r r r r r3 r- r( rT listr+ r* r4 r" r" r# test_serialize_ints_to_strings s
z'TestGEXF.test_serialize_ints_to_stringsc C s t }|dddg tdD ]N}||j| d< ||j| d< ||j| d< ||j| d< |d |j| d
< q d td d
t j d}dt |}||kst
d S )N)r r r r
r
rP r r pidstartr end
NetworkX a
) r r add_edges_fromranger* timestrftime __version__join
generate_gexfr+ )r/ r0 iexpectedobtainedr" r" r# test_write_with_node_attributes s z(TestGEXF.test_write_with_node_attributesc C sf t }|ddddifddddifdg dtd dt j d }d
t |}||ksbtd S ) Nr r r r
rZ r_ r` ra a
rb ) r r rc re rf rg rh ri r+ )r/ r0 rk rl r" r" r# test_edge_id_construct8 s $z TestGEXF.test_edge_id_constructc C s t d}td}t|dd |dD d |d|d d d < d
td d tj d
}d
t |}||ks~t d S )NZnumpyr[ c S s i | ]
}||qS r" r" )r: nr" r" r#
W s z,TestGEXF.test_numpy_type..numberg?r r z edge-numberr_ r` ra a_
rb )
rE Z importorskipr Z
path_graphZset_node_attributesZarangeZfloat64re rf rg rh ri r+ )r/ npr0 rk rl r" r" r# test_numpy_typeT s
/zTestGEXF.test_numpy_typec C sX t }|jddd t }t || |d t j|td}|j d d sTt
d S )Nr T)testattrr rR rt r r r r r r3 r- r( rT r* r+ r4 r" r" r# test_bool s
zTestGEXF.test_boolc C s ddl m} tdtd }}t }|jd|ddd |jd|dd d |jd
| d d d t }t|| | d |
}| d tj |t d
}d|kst
d|kst
d|kst
|jd d |kst
||jd d st
|jd
d | kst
|jd d dks t
|jd d dks6t
|jd
d d ksNt
|jd d dksft
|jd d d ks~t
|jd
d d kst
d S )Nr )isnaninfnanr rA )rt strdatakeyr
rB rP z-infcrR s INFs NaNs -INFrt rz networkx_key)mathrw floatr r r r r r3 r- readr( rT r+ r* )r/ rw rx ry r0 r5 Zfiletextr1 r" r" r#
test_specials s0
zTestGEXF.test_specialsc C sd t }ddg}|jd|d t }t || |d t j|td}|j d d |ks`t
d S )N)r r
rP ) r r
r )r{ r rR r} ru )r/ r0 Z
list_valuer5 r1 r" r" r# test_simple_list s
zTestGEXF.test_simple_listc C s t }|jdddd d|jd< t }t || |d t j|t d}t
| t
| ksjt t
d d
|
D t
d d
|
D kst d S ) Nr r rM rN ZdynamicrS r rR c s s | ]}t |V qd S r7 r8 r9 r" r" r# r< s z-TestGEXF.test_dynamic_mode..)r r r r r r r3 r- r( rT r) r* r+ r, r4 r" r" r# test_dynamic_mode s
zTestGEXF.test_dynamic_modec C s t }|jdddd |jdddd |jddddd ddd
|jdddd ddd |dd t }t || |d t j|t d
}t
| t
| kst t
dd |
D t
dd |
D kst d S )Nr r rM rN r r r rP Z
undirected)r Zweighttyper] r^ r )r r r] r^ rR c s s | ]}t |V qd S r7 r8 r9 r" r" r# r< s zCTestGEXF.test_multigraph_with_missing_attributes..)r Z
MultiGraphr r r r r3 r- r( rT r) r* r+ r, r4 r" r" r# 'test_multigraph_with_missing_attributes s
z0TestGEXF.test_multigraph_with_missing_attributesc C sR t }|jdddd ddi|jd d< dddd |jd d d
< ddd d |jd d d
< d|jd d d< d|jd d d< t }t j||dd |d t j|t d}t
| t
| kst t
dd | D t
dd | D kst t }t j||dd |d t j|t d}|jd d d
d dksNt t }|jdddd ddi|jd d< dddd |jd d d
< ddd dd|jd d d
< d |jd d d< d|jd d d< t }t || |d t j|t d}t
| t
| ks t t
d d | D t
d d | D ksNt d S ) Nr r rM rN size6 Zvizr )xyzposition )rgrB rO zhttp://random.urlshaper
Z thickness1.1draftversionrR c s s | ]}t |V qd S r7 r8 r9 r" r" r# r< s z7TestGEXF.test_missing_viz_attributes..z1.2draftrA g ?g ?)r r rB rA zftp://random.urlc s s | ]}t |V qd S r7 r8 r9 r" r" r# r< s
r r r r* r r r3 r- r( rT r) r+ r, r4 r" r" r# test_missing_viz_attributes sF
z$TestGEXF.test_missing_viz_attributesc C sL t }|jdddd dg|jd d< t }t || |d t j|t d}t
| t
| kspt t
dd | D t
dd | D kst t }|jdddd dg|jd d
< t }t j||d d |d t j|t d}t
| t
| kst t
d
d | D t
d
d | D ksHt d S )Nr r rM rN rY ZspellsrR c s s | ]}t |V qd S r7 r8 r9 r" r" r# r< s z0TestGEXF.test_slice_and_spell..Zslicesr r c s s | ]}t |V qd S r7 r8 r9 r" r" r# r< s r r4 r" r" r# test_slice_and_spell s,
z TestGEXF.test_slice_and_spellc C s t }|jdddddgd t }t || |d t j|td}t |
t |
ksft t dd | D t dd | D kst d S )
Nr r rM r r
)r rO parentsrR c s s | ]}t |V qd S r7 r8 r9 r" r" r# r< + s z+TestGEXF.test_add_parent..)
r r r r r r3 r- r( rT r) r* r+ r, r4 r" r" r# test_add_parent# s
zTestGEXF.test_add_parentN)__name__
__module__ __qualname__ classmethodr$ r2 r6 r= rC rH rI rJ rL rU rX rm rn rs rv r r r r r r r r" r" r" r# r s.
|
# 8
,r )r re rE Znetworkxr r r" r" r" r# s