o
]ZhK @ 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" Q/var/www/auris/lib/python3.10/site-packages/networkx/readwrite/tests/test_gexf.py setup_class
sN
8
zTestGEXF.setup_classc C s~ | j }t| j}t| t| ksJ t| t| ks%J t|jddt|jddks7J | 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| ks%J t| t| ks3J t|j ddt|j ddksEJ | 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| ksJ tdd | D tdd | D ks/J | jd d S )Nc 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|jddksJ t|jdd}t|jdd}t||D ]
\}}||ks9J 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 * d}t |d}ttjtj| d S )Na
r r r r pytestZraisesr Z
NetworkXErrorr( r. sr4 r" r" r# test_directed_edge_in_undirected
z)TestGEXF.test_directed_edge_in_undirectedc C rE )Na
r rF rH r" r" r# test_undirected_edge_in_directed rK z)TestGEXF.test_undirected_edge_in_directedc C rE )Na
r rF rH 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. rI 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 |
ksNJ t dd | D t dd | D ksfJ |jd= |j|jksrJ d S )Nr r greenr color)r r r
r
rS )r rR yellowr
r Z edge_defaultr Z node_typec s r6 r7 r8 r9 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 |dgks+J |j
d d dks6J d S ) Nr rU M r r rV r Z77) r r r r r r2 r, r( rX listr* r3 r" r" r# test_serialize_ints_to_strings s
z'TestGEXF.test_serialize_ints_to_stringsc C s t }|g d tdD ]'}||j| d< ||j| d< ||j| d< ||j| d< |d |j| d< qd td
d t j d }d
t |}||ksRJ d S )N))r r r r
r
rS 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 |}||ks1J d S ) Nr r r r
r^ rc rd re a
rf )r r rg ri rj rk rl rm )r. r/ ro rp 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" )r: nr" r" r#
W s z,TestGEXF.test_numpy_type..numberg?r r z edge-numberrc rd re a_
rf ) rG Z importorskipr Z
path_graphZset_node_attributesZarangeZfloat64ri rj rk rl rm )r. npr/ ro rp r" r" r# test_numpy_typeT s
/zTestGEXF.test_numpy_typec C sX t }|jddd t }t || |d t j|td}|j d d s*J d S )Nr T)testattrr rV ry
r r r r r r2 r, r( rX 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 sTJ d|v sZJ d|v s`J |j
d d |kskJ ||j
d d svJ |j
d
d | ksJ |j
d d dksJ |j
d d dksJ |j
d
d d ksJ |j
d d dksJ |j
d d d ksJ |j
d
d d ksJ d S )Nr )isnaninfnanr rB )ry strdatakeyr
rC rS z-infcrV s INFs NaNs -INFry r networkx_key)mathr| floatr r r r r r2 r, readr( rX r* )r. r| r} 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 |ks0J d S )N)r r
rS ) r r
r )r r rV r rz )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
| ks5J t
d d
| D t
d d
| D ksMJ d S ) Nr r rP rQ ZdynamicrW r rV c s r6 r7 r8 r9 r" r" r# r< r= z-TestGEXF.test_dynamic_mode..)
r r r r r r r2 r, r( rX 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
| ksUJ t
dd | D t
dd | D ksmJ d S )Nr r rP rQ r r r rS Z
undirected)r weighttypera rb r )r r ra rb rV c s r6 r7 r8 r9 r" r" r# r< r= zCTestGEXF.test_multigraph_with_missing_attributes..)
r Z
MultiGraphr r r r r2 r, r( rX r) r* r+ r3 r" r" r# 'test_multigraph_with_missing_attributes s
z0TestGEXF.test_multigraph_with_missing_attributesc C sP 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
| ksgJ t
dd | D t
dd | D ksJ t }t j||dd |d t j|t d}|jd d d
d dksJ 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 ks&J d S ) Nr r rP rQ size6 Zvizr )xyzposition )rgrC rR zhttp://random.urlshaper
Z thickness1.1draftversionrV c s r6 r7 r8 r9 r" r" r# r< r= z7TestGEXF.test_missing_viz_attributes..z1.2draftrB g ?g ?)r r rC rB zftp://random.urlc s r6 r7 r8 r9 r" r" r# r< r= r r r r* r r r2 r, r( rX r) r+ r3 r" r" r# test_missing_viz_attributes sF
z$TestGEXF.test_missing_viz_attributesc C sH t }|jdddd dg|jd d< t }t || |d t j|t d}t
| t
| ks8J t
dd | D t
dd | D ksPJ 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 ksJ d S )Nr r rP rQ r] ZspellsrV c s r6 r7 r8 r9 r" r" r# r< r= z0TestGEXF.test_slice_and_spell..Zslicesr r c s r6 r7 r8 r9 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 |
ks3J t dd | D t dd | D ksKJ d S )
Nr r rP r r
)r rR parentsrV c s r6 r7 r8 r9 r" r" r# r< + r= z+TestGEXF.test_add_parent..) r r r r r r2 r, r( rX r) r* r+ r3 r" r" r# test_add_parent# s
zTestGEXF.test_add_parentN)__name__
__module__ __qualname__ classmethodr$ r1 r5 r> rD rJ rL rM rO rY r\ rq rr rx r{ r r r r r r r r" r" r" r# r s0
|
# 8
, r )r ri rG Znetworkxr r r" r" r" r# s