a
hK @ 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" P/var/www/auris/lib/python3.9/site-packages/networkx/readwrite/tests/test_gexf.py setup_class
sN
8
zTestGEXF.setup_classc C s~ | j }t| j}t| t| ks.J t| t| ksJJ t|jddt|jddksnJ | jd d S NTdatar )r r read_gexfr sortednodesedgesseekselfGHr" r" r# !test_read_simple_directed_graphml s $z*TestGEXF.test_read_simple_directed_graphmlc C s | j }t }t|| |d t|}t| t| ksJJ t| t| ksfJ t|j ddt|j ddksJ | j
d d S )Nr Tr& ) r r r r
write_gexfr, r( r) r* r+ r r. r/ fhr0 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.J tdd | D tdd | D ks^J | jd d S )Nc s s | ]}t |V qd S Nr) .0er" r" r# z?TestGEXF.test_read_simple_undirected_graphml..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|jddks4J t|jdd}t|jdd}t||D ]\}}||ks^J q^| jd d S r% ) r r r( r r) r* r+ zipr, )r. r/ r0 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. sr4 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:J d S )Nay
r T)Zrelabelr Word)r r r r r( r) r* )r. rG r4 r/ r" r" r# test_relabel s
zTestGEXF.test_relabelc C s t }|jdddd t |g d |jdddd d d
i|jd < d d
i|jd< t }t || | d t j
|t d}t |
t |
ksJ t dd | D t dd | D ksJ |jd= |j|jksJ d S )Nr r greenr color)r r r
r
rP )r rO Zyellowr
r Z edge_defaultr Z node_typec s s | ]}t |V qd S r6 r7 r8 r" r" r# r; r< z2TestGEXF.test_default_attribute..mode)r r r Zadd_pathr r r r r2 r, r( intr) r* r+ r3 r" r" r# test_default_attribute s
z TestGEXF.test_default_attributec C sp t }|jdddd t }t || |d t j|td}t |dgksVJ |j
d d dkslJ d S ) Nr rQ M r r rR r Z77) r r r r r r2 r, r( rT listr* r3 r" r" r# test_serialize_ints_to_strings s
z'TestGEXF.test_serialize_ints_to_stringsc C s t }|g d tdD ]N}||j| d< ||j| d< ||j| d< ||j| d< |d |j| d< q d td
d t j d }d
t |}||ksJ d S )N))r r r r
r
rP r r pidstartr end
NetworkX a
)
r r add_edges_fromranger* timestrftime __version__join
generate_gexf)r. r/ 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 |}||ksbJ d S ) Nr r r r
rZ r_ r` ra a
rb )r r rc re rf rg rh ri )r. r/ 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~J d S )Nnumpyr[ c S s i | ]
}||qS r" r" )r9 nr" r" r#
W r< z,TestGEXF.test_numpy_type..numberg?r r z edge-numberr_ r` ra a_
rb ) rE Z importorskipr Z
path_graphZset_node_attributesZarangefloat64re rf rg rh ri )r. npr/ 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 sTJ d S )Nr T)testattrr rR rv
r r r r r r2 r, r( rT r* r3 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|v sJ d|v sJ d|v sJ |j
d d |ksJ ||j
d d sJ |j
d
d | ksJ |j
d d dks J |j
d d dks6J |j
d
d d ksNJ |j
d d dksfJ |j
d d d ks~J |j
d
d d ksJ d S )Nr )isnaninfnanr rA )rv strdatakeyr
rB rP z-infcrR s INFs NaNs -INFrv r| networkx_key)mathry floatr r r r r r2 r, readr( rT r* )r. ry rz r{ r/ r4 Zfiletextr0 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`J d S )N)r r
rP ) r r
r )r} r rR r rw )r. r/ Z
list_valuer4 r0 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
| ksjJ t
d d
| D t
d d
| D ksJ d S ) Nr r rM rN ZdynamicrS r rR c s s | ]}t |V qd S r6 r7 r8 r" r" r# r; r< z-TestGEXF.test_dynamic_mode..)
r r r r r r r2 r, r( rT r) r* r+ r3 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
| ksJ t
dd | D t
dd | D ksJ 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 r6 r7 r8 r" r" r# r; r< zCTestGEXF.test_multigraph_with_missing_attributes..)
r Z
MultiGraphr r r r r2 r, r( rT r) r* r+ r3 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
| ksJ t
dd | D t
dd | D ksJ t }t j||dd |d t j|t d}|jd d d
d dksNJ 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 J t
d d | D t
d d | D ksNJ d S ) Nr r rM rN size6 Zvizr )xyzposition )rgrB rO zhttp://random.urlshaper
Z thickness1.1draftversionrR c s s | ]}t |V qd S r6 r7 r8 r" r" r# r; r< z7TestGEXF.test_missing_viz_attributes..z1.2draftrA g ?g ?)r r rB rA zftp://random.urlc s s | ]}t |V qd S r6 r7 r8 r" r" r# r; r< r r r r* r r r2 r, r( rT r) r+ r3 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
| kspJ t
dd | D t
dd | D ksJ t }|jdddd dg|jd d
< t }t j||d d |d t j|t d}t
| t
| ksJ t
d
d | D t
d
d | D ksHJ d S )Nr r rM rN rY ZspellsrR c s s | ]}t |V qd S r6 r7 r8 r" r" r# r; r< z0TestGEXF.test_slice_and_spell..Zslicesr r c s s | ]}t |V qd S r6 r7 r8 r" r" r# r; r< r r3 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 |
ksfJ t dd | D t dd | D ksJ d S )
Nr r rM r r
)r rO parentsrR c s s | ]}t |V qd S r6 r7 r8 r" r" r# r; + r< z+TestGEXF.test_add_parent..) r r r r r r2 r, r( rT r) r* r+ r3 r" r" r# test_add_parent# s
zTestGEXF.test_add_parentN)__name__
__module__ __qualname__ classmethodr$ r1 r5 r= rC rH rI rJ rL rU rX rm rn ru rx 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