o
]Zh @ s d dl Z d dlZd dlZd dlmZ d dlmZmZ G dd dZ G dd de Z
G dd d e Z G d
d d e Z d d
Z
dd Zdd Zdd ZdS ) N)
GraphMLWriter) edges_equal nodes_equalc @ s e Zd Zedd ZdS ) BaseGraphMLc C s4 d| _ t | _| jd | jjdddd | jdd | jg d t| j d| _
d | _ tjd
d| _ d d i| j j
d
< | j jddd | j jddd | j jddd | j d | j jddd | j jddddd | j jddddd | j jdddd d | j jddd d | j jddd d | j jddd d | j jddd d!d t| j d| _d"| _tjd
d| _d#d$d$d%d%d&d'| jj
d
< | jd | jd | jjdddd t| j d| _d(| _t | _| jjd)d*d+d, | jjd-d.d+d, | jjd)d-d/d0 t| j d}|| _d1| _t | _| jjdd2d3 | jjdd d3 | jjddd2d3 | jjdddd3 t| j d}|| _d4| _t | _| jd | jjdddd | jd5d6g t| j d}|| _d7| _ t | _ | j d | j jdddd | j jdddd | j jdddd t| j d}|| _ d8| _ t | _!| j!d | j!jdddd | j!jdddd | j!jdddd t| j d}|| _"d9| _#t | _$| j$d | j$dd | j$jdddd | j$jdddd t| j# d}|| _%d S ):Na
Zn10n0n2fooid)
n1r r n3)r n5)r n4)r n6)r r )r n7)r n8)r r )r Zn9UTF-8a
yellow
green
blue
red
turquoise
1.0
1.0
2.0
1.1
Gcoloryellow node_defaultgreen)r bluer redr r Z turquoisee0g ?)r
weightr e1e2g @Ze3Ze4Ze5Ze6皙?a
false
0
0
0.0
0.0
Foo
Fr g ZFoo)Zboolean_attributeZ
int_attributeZlong_attributeZfloat_attributeZdouble_attributeZstring_attributeaV
val1
val2
val_one
val2
edge_value
0Zval1Zval2)prop1prop21Zval_oneZ
edge_value) edge_propaC
1
2.0
1
k
1.0
r a
r r
a
a
a
)&simple_directed_datanxZDiGraphsimple_directed_graphadd_nodeadd_edgeadd_edges_fromioBytesIOencodesimple_directed_fhattribute_dataattribute_graphgraph attribute_fhZnode_attribute_default_data node_attribute_default_graphnode_attribute_default_fhZ attribute_named_key_ids_data attribute_named_key_ids_graphattribute_named_key_ids_fhZattribute_numeric_type_data attribute_numeric_type_graphattribute_numeric_type_fhsimple_undirected_dataGraphsimple_undirected_graphsimple_undirected_fhundirected_multigraph_data
MultiGraphundirected_multigraphundirected_multigraph_fh'undirected_multigraph_no_multiedge_data"undirected_multigraph_no_multiedge%undirected_multigraph_no_multiedge_fh'multigraph_only_ids_for_multiedges_data"multigraph_only_ids_for_multiedges%multigraph_only_ids_for_multiedges_fh)clsfh rL T/var/www/auris/lib/python3.10/site-packages/networkx/readwrite/tests/test_graphml.py setup_class s
*
zBaseGraphML.setup_classN)__name__
__module__ __qualname__ classmethodrN rL rL rL rM r
s r c @ s e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%S )&TestReadGraphMLc C s | j }t| j}t| t| ksJ t| t| ks%J t|jddt|jddks7J | jd t| j }t| t| ksQJ t| t| ks_J t|jddt|jddksqJ d S NTdatar )
r* r) read_graphmlr1 sortednodesedgesseek
parse_graphmlr( selfr HPGrL rL rM !test_read_simple_directed_graphml: s $ (z1TestReadGraphML.test_read_simple_directed_graphmlc C | j }t| j}t| | sJ t| | s J | jd t | j
}t| | s6J t| | sAJ d S Nr ) r> r) rW r? r rY r rZ r[ r\ r< r] rL rL rM #test_read_simple_undirected_graphmlG z3TestReadGraphML.test_read_simple_undirected_graphmlc C rb rc ) rB r) rW rC r rY r rZ r[ r\ r@ r] rL rL rM 'test_read_undirected_multigraph_graphmlR re z7TestReadGraphML.test_read_undirected_multigraph_graphmlc C rb rc ) rE r) rW rF r rY r rZ r[ r\ rD r] rL rL rM 4test_read_undirected_multigraph_no_multiedge_graphml] re zDTestReadGraphML.test_read_undirected_multigraph_no_multiedge_graphmlc C rb rc ) rH r) rW rI r rY r rZ r[ r\ rG r] rL rL rM ?test_read_undirected_multigraph_only_ids_for_multiedges_graphmlh re zOTestReadGraphML.test_read_undirected_multigraph_only_ids_for_multiedges_graphmlc C s | j }t| j}t|dt|jddsJ t|jdd}t|jdd}t||D ]
\}}||ks8J q.| j d t
| j }t|dt|jddksVJ t|jdd}t|jdd}t||D ]
\}}||ksuJ qkd S rT ) r3 r) rW r5 r rY rX rZ zipr[ r\ r2 )r^ r r_ geheabr` rL rL rM test_read_attribute_graphmls s "z+TestReadGraphML.test_read_attribute_graphmlc C s. | j }t| j}|jd |jd ksJ d S )Nr )r6 r) rW r7 r4 )r^ r r_ rL rL rM #test_node_default_attribute_graphml s z3TestReadGraphML.test_node_default_attribute_graphmlc C < d}t |d}ttjtj| ttjtj| d S )Na
r r. r/ r0 pytestraisesr)
NetworkXErrorrW r\ r^ srK rL rL rM test_directed_edge_in_undirected
z0TestReadGraphML.test_directed_edge_in_undirectedc C rp )Na
r rq ru rL rL rM test_undirected_edge_in_directed rx z0TestReadGraphML.test_undirected_edge_in_directedc C rp )Na
yellow
green
blue
1.0
r rq ru rL rL rM test_key_raise s z TestReadGraphML.test_key_raisec C rp )Na5
yellow
green
blue
r rq ru rL rL rM test_hyperedge_raise s z$TestReadGraphML.test_hyperedge_raisec C sn d}t |d}t|}ddg}t|jdd|ks J |d t|}t|jdd|ks5J d S )Na
r )r r r )r r r Tkeysr ) r. r/ r0 r) rW rX rZ r[ r\ r^ rv rK r expectedr_ rL rL rM test_multigraph_keys s
z$TestReadGraphML.test_multigraph_keysc C sH t }|d |d |dddddifddddifdddd d
fddd dd ifddd
ddifdddddd
fg t }t || |d t j|t d}t
|j ddd|j dddsdJ |j |j kslJ dd |j
D }|d t j|ttd}||j ksJ |d | }t j|ttd}||j ksJ dS )zf
Test that data and keys of edges are preserved on consequent
write and reads
r& )r& r keyZ data_key1r
Zdata_id2Z data_key3Zdata_id3)r r
g Z data_key4h Zdata_id5i Z data_key6Zdata_id7r node_typeT)rV r} c S ( i | ]\}}t |d d | D qS )c S r )c S s i | ] \}}t ||qS rL str).0ZekeyddrL rL rM
s zWTestReadGraphML.test_preserve_multi_edge_data....r items)r ZnbrZkey_dictrL rL rM r s zLTestReadGraphML.test_preserve_multi_edge_data...r )r nodeZnbr_dictrL rL rM r s zATestReadGraphML.test_preserve_multi_edge_data..)r Z
edge_key_typeN)r) rA r+ r- r. r/
write_graphmlr[ rW intr rZ _adjr r readr\ )r^ r rK r_ ZGadjZHHZ string_fhrL rL rM test_preserve_multi_edge_data s:
"
z-TestReadGraphML.test_preserve_multi_edge_datac C s d}t |d}tj|dd}t| dgks J |jdddd s&J |jd d
d ks1J |jd d
d ks
1
2
3
r TZforce_multigraph)r r r r r )r labelr$ 2r 3Z
shape_typeZ rectanglez com.yworks.flowchart.terminator descriptionzdescription
line1
line2r r
) r. r/ r0 r) rW listrZ Zhas_edgerY r[ r\ )r^ rV rK r r_ rL rL rM test_yfiles_extension+ sF Z
z%TestReadGraphML.test_yfiles_extensionc C s d}t |d}t|}t|}||fD ]8}|jd d s#J |jd d r,J |jd d r5J |jd d s>J |jd d rGJ |jd d sPJ qd S )
NaU
false
true
false
FaLsE
True
0
1
r r testr r r r r )r. r/ r0 r) rW r\ rY )r^ rv rK r r_ r4 rL rL rM test_bool s
zTestReadGraphML.test_boolc C s d}d}d}||fD ]$}t |d}t|}t|}||fD ] }|jd d s-J q"q
t |d}ttj tj| ttj tj| d S )Na
false
true
a
false
true
a%
false
true
r r r )
r. r/ r0 r) rW r\ rY rr rs rt ) r^ ZgoodbadZuglyrv rK r r_ r4 rL rL rM test_graphml_header_line s
z(TestReadGraphML.test_graphml_header_linec C sb d}t |d}t|}dd |jddD }t|dks"J |D ]
}|d d ks.J q$d S )
NaI
2
Group 3
Folder 3
Group 1
Folder 1
1
3
Group 2
Folder 2
5
6
9
r c S s g | ]\}}|qS rL rL )r _xrL rL rM
( zDTestReadGraphML.test_read_attributes_with_groups..TrU ZCustomProperty )r. r/ r0 r) rW rY len)r^ rV rK r Z node_datarL rL rM test_read_attributes_with_groups s
z0TestReadGraphML.test_read_attributes_with_groupsc C st d}t |d}t|}dddifg}t|jdd|ks"J |d t|}t|jdd|ks8J d S ) Na
4284
r r Z cudfversioni TrU r ) r. r/ r0 r) rW rX rY r[ r\ r~ rL rL rM test_long_attribute_type- s
z(TestReadGraphML.test_long_attribute_typeN)rO rP rQ ra rd rf rg rh rn ro rw ry rz r{ r r r r r r r rL rL rL rM rS 9 s,
- +3 &rS c @ s e Zd ZeejZedd Zdd Z dd Z
dd Z d d
Z d d Z
d
d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)S )*TestWriteGraphMLc C s t td}d S )N
lxml.etree)r rN rr importorskip)rJ r rL rL rM rN H s z TestWriteGraphML.setup_classc C sD zdd l }tjtjks
J W d S ty! tjtjks J Y d S w rc )Z
lxml.etreer) r write_graphml_lxml ImportErrorwrite_graphml_xml)r^ ZlxmlrL rL rM test_write_interfaceM s z%TestWriteGraphML.test_write_interfacec C s | j }d|jd< t }| || |d t|}t| t| ks*J t|
t|
ks8J t|j
ddt|j
ddksJJ | j d d S )Ntherehir TrU r* r4 r. r/ writerr[ r) rW rX rY rZ r1 r^ r rK r_ rL rL rM 'test_write_read_simple_directed_graphmlU s
$z8TestWriteGraphML.test_write_read_simple_directed_graphmlc C s& t }| j}| }|||g d S N)r r* copyZ
add_graphs)r^ Zgmlwr r_ rL rL rM test_GraphMLWriter_add_graphsa s z.TestWriteGraphML.test_GraphMLWriter_add_graphsc C s | j }d|jd< d|jd< t }| j||dd |d t|}t| t| ks1J t|
t|
ks?J t|j
dd t|j
dd ksQJ | j d d S )
Nr r r$ r
F)Z prettyprintr TrU r r rL rL rM %test_write_read_simple_no_prettyprintg s
$z6TestWriteGraphML.test_write_read_simple_no_prettyprintc C s ddl m} | j}t }| j||dd |d t|}|d t |
|
s/J t | | s:J t |j dd|j ddsIJ | j
d ||}t| }t|dksaJ dd |d d D }t|d kstJ d
|d v s|J d |d v sJ d |d
v sJ d|d
v sJ d|d v sJ d|d v sJ t }t|| |d t|}t } tj|| dd | d t| }
tdd t|j
|
j
D sJ tdd t|j |
j D sJ d S )Nr parseT)Z
named_key_idsrU c S g | ]}| qS rL r r childrL rL rM r r zTTestWriteGraphML.test_write_read_attribute_named_key_ids_graphml.. )r
r% ) attr.namer% )r
r# r& )r r# )r
r" r )r r" c s | ] \}}||kV qd S r rL )r r r rL rL rM zSTestWriteGraphML.test_write_read_attribute_named_key_ids_graphml..c s r r rL )r r r rL rL rM r r )xml.etree.ElementTreer r8 r. r/ r r[ r) rW r rY r rZ r9 r getrootr r allri ) r^ r r rK r_ xmlchildrenr} Zdefault_behavior_fhZnamed_key_ids_behavior_fhJrL rL rM /test_write_read_attribute_named_key_ids_graphmlt s@
$z@TestWriteGraphML.test_write_read_attribute_named_key_ids_graphmlc C s ddl m} | j}t }| j||dd |d t|}|d t |
|
s/J t | | s:J t |j dd|j ddsIJ | j
d ||}t| }t|dksaJ dd |d d D }t|d kstJ d
|d v s|J d
|d v sJ d S ) Nr r TZinfer_numeric_typesrU r c S r rL r r rL rL rM r r zSTestWriteGraphML.test_write_read_attribute_numeric_type_graphml..r )z attr.typedoubler& )r r r: r. r/ r r[ r) rW r rY r rZ r; r r r )r^ r r rK r_ r r r} rL rL rM .test_write_read_attribute_numeric_type_graphml s$
z?TestWriteGraphML.test_write_read_attribute_numeric_type_graphmlc C sr t }|ddg |d }| || t |}| s J t|jdd|jdds/J |j|jks7J dS )zWriting keys as edge id attributes means keys become strings.
The original keys are stored as data, so read them back in
if `str(key) == edge_id`
This allows the adjacency to remain the same.
rl rm r )rl rm r test.graphmlTr| N) r) rA r- r rW
is_multigraphr rZ r r^ tmp_pathr fnamer_ rL rL rM test_more_multigraph_keys s
z*TestWriteGraphML.test_more_multigraph_keysc C s t jdd}|jdddd t |g d |jdddd d
d i|jd < d
di|jd< t }| || | d t j
|t d}t |
|
sMJ t| | sXJ |j|jks`J d S )NZFred)namer& r )r r )r r& r r r r r' r r r r Z edge_defaultr r )r) r= r+ Zadd_pathr, r4 r. r/ r r[ rW r r rY r rZ r rL rL rM test_default_attribute s
z'TestWriteGraphML.test_default_attributec C s t }|jddd |jddd |jdddd |jdddd t }| || |d t |}|j d d r?J |j d d dksJJ |j
d d rSJ |j
d d dks^J d S ) Nr Fspecialr r r r r r r r r& r) rA r+ r, r. r/ r r[ rW rY rZ r rL rL rM test_mixed_type_attributes s
z+TestWriteGraphML.test_mixed_type_attributesc C s t }|jddd |jddd |jdddd |jdddd t }| || |d t |}|j d d dksAJ |j d d dksLJ |j
d d dksWJ |j
d d dksbJ d S ) Nr Zhellor r r r r r r r rL rL rM %test_str_number_mixed_type_attributes s
z6TestWriteGraphML.test_str_number_mixed_type_attributesc C s t d}t }|jd|dd |jddd |jdd|dd |jdddd t }| || |
d t |}|j d d dksLJ |j d d dksWJ |j
d
d dksbJ |j
d d dksmJ d S ) Nnumpyr r r r r& r r r r r )rr r r) rA r+ Zint64r, r. r/ r r[ rW rY rZ )r^ npr rK r_ rL rL rM %test_mixed_int_type_number_attributes s
z6TestWriteGraphML.test_mixed_int_type_number_attributesc C s t }|ddg |d }| || t |}| r J t j|dd}| s-J |ddd |d }| || t |}| sIJ t j|dd}| sVJ d S ) Nr )rm cr r Tr rl rm ze-id)r) rA r- r rW r r, r rL rL rM test_multigraph_to_graph s
z)TestWriteGraphML.test_multigraph_to_graphc
C s ddl m} t }|g d dd |jD }t||d |d }| j||dd tj|dd}t |}t
| | sCJ t | | sNJ t||d t |jd
d |jd
d sdJ ||}t
| } t| d ksvJ d
d | dD }
t|
t| ksJ d|} t| } t| t| ksJ t| t| ksJ t||d t |jd
d | jd
d sJ d S )Nr r )rl rm rm r rl r c S i | ]}|t |qS rL r r erL rL rM r zOTestWriteGraphML.test_write_generate_edge_id_from_attribute..eidr Zedge_id_from_attributer
TrU r c S g | ]}|j d qS r Zattribr edgerL rL rM r 2 zOTestWriteGraphML.test_write_generate_edge_id_from_attribute....//{http://graphml.graphdrawing.org/xmlns}edger )r r r) r= r- rZ set_edge_attributesr generate_graphmlrW r rY r r r r findallrX valuesjoinr\ )
r^ r r r edge_attributesr generatorr_ treer edge_idsrV r rL rL rM *test_write_generate_edge_id_from_attribute s8
"z;TestWriteGraphML.test_write_generate_edge_id_from_attributec
C s ddl m} t }|g d dd |jD }t||d |d }| j||dd tj|dd}t |}|
s>J tj |d d
}|
sKJ t | | sVJ t
| | saJ td d |jd d
D t| ksvJ tdd |jd dD t| ksJ ||}t| } t| dksJ dd | dD }
t|
t| ksJ d|} t| } |
sJ t | | sJ t
| | sJ tdd | jd d
D t| ksJ tdd | jd dD t| ksJ d S )Nr r )r r r r c S r rL r r rL rL rM r I r zZTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..r r r Tr c s | ] \}}}| d V qdS r Ngetr uvrV rL rL rM r X zYTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..rU c s | ]\}}}|V qd S r rL r r r r rL rL rM r \ r| r c S r r r r rL rL rM r c r zZTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..r r c s r r r r rL rL rM r s r c s r r rL r rL rL rM r w r
)r r r) rA r- rZ r r r rW r r rY r rX r r r r r r r\ )
r^ r r r r r r r_ r r r Z graphml_datar rL rL rM 5test_multigraph_write_generate_edge_id_from_attributeD sP
zFTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attributec C t d}|d}tddd|ifg}|d }| || tj|td}|j|jks.J |d d d }|d d d }|t j |dd ksIJ t
||jksRJ t
|t ksZJ d S
Nr g333333 @r& r r r r gư>)abs) rr r float64r) r= r rW r rZ approxtypefloat r^ r r wtr r r_ ZwtGZwtHrL rL rM test_numpy_float64{
z#TestWriteGraphML.test_numpy_float64c C r r
) rr r Zfloat32r) r= r rW r rZ r r r r rL rL rM test_numpy_float32 r z#TestWriteGraphML.test_numpy_float32c C sZ t d}| j}|d|jd d< |d }| j||dd t|}|j|jks+J d S )Nr r )r r r r Tr ) rr r r: r rZ r r) rW r )r^ r r r r r_ rL rL rM test_numpy_float64_inference s
z-TestWriteGraphML.test_numpy_float64_inferencec C s t }tdtd td }tdtd td }t}|j|d|d |d }| || t j||d
}|j|jks?J d S ) Ni( { i i i iD Z Radiohead)r r r )r) r= chrr r, r rW r )r^ r r Zname1Zname2r r r_ rL rL rM test_unicode_attributes s z(TestWriteGraphML.test_unicode_attributesc C sp dd l }ddi}||}t }||jd< t }| || |d t |}|jd |jd ks6J d S )Nr rl z {"a": "123"}r )
jsondumpsr) r= r4 r. r/ r r[ rW )r^ r rl sar rK r_ rL rL rM test_unicode_escape s
z$TestWriteGraphML.test_unicode_escapeN) rO rP rQ staticmethodr) r r rR rN r r r r r r r r r r r r r r r r r r r rL rL rL rM r E s.
-)7 r c @ s" e Zd ZeejZedd ZdS )TestXMLGraphMLc C s t d S r )r rN )rJ rL rL rM rN s zTestXMLGraphML.setup_classN) rO rP rQ r r) r r rR rN rL rL rL rM r! s
r! c C sl t d t } | jdg dd t }t jtdd t | | W d dS 1 s/w Y dS )qTest that a detailed exception is raised when an attribute is of a type
not supported by GraphML, e.g. a listr r r r& r my_list_attributeGraphML does not supportmatchN)
rr r r) r= r+ r. r/ rs TypeErrorr r rK rL rL rM 1test_exception_for_unsupported_datatype_node_attr s
"r+ c C sn t d t } | jddg dd t }t jtdd t | | W d dS 1 s0w Y dS ) r" r r r& r# r$ r& r' N)
rr r r) r= r, r. r/ rs r) r r* rL rL rM 1test_exception_for_unsupported_datatype_edge_attr s
"r, c C sh t d t } g d| jd< t }t jtdd t | | W d dS 1 s-w Y dS )r" r r# r% r& r' N)
rr r r) r= r4 r. r/ rs r) r r* rL rL rM 2test_exception_for_unsupported_datatype_graph_attr s
"r- c C sR d} t | d}t|}|jd dddksJ |jd dd dks'J d
S ) zPTests that a GraphML string with an empty attribute can be parsed
correctly.a
aaa
bbb
ccc
r r! Zaaabbb)r barr$ cccr N)r. r/ r0 r) rW rY )rv rK r rL rL rM test_empty_attribute s
r1 )r. rr Znetworkxr) Znetworkx.readwrite.graphmlr Znetworkx.utilsr r r rS r r! r+ r, r- r1 rL rL rL rM s. 1 w