o
˷e @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlm Z m
Z
G dd dZ G dd de Z G dd d e Z
G d
d d e
Zd d
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
n10n0n2fooid)
n1r r n3)r n5)r n4)r n6)r r )r n7)r n8)r r )r n9UTF-8a
yellow
green
blue
red
turquoise
1.0
1.0
2.0
1.1
Gcoloryellow node_defaultgreen)r bluer redr r turquoisee0g ?)r weightr
e1e2g @e3e4e5e6皙?a
false
0
0
0.0
0.0
Foo
Fr g Foo)boolean_attribute
int_attributelong_attributefloat_attributedouble_attributestring_attributeaV
val1
val2
val_one
val2
edge_value
0val1val2)prop1prop21val_one
edge_value) edge_propaC
1
2.0
1
k
1.0
r a
r r a
a
a
)&simple_directed_datanxDiGraphsimple_directed_graphadd_nodeadd_edgeadd_edges_fromioBytesIOencodesimple_directed_fhattribute_dataattribute_graphgraph attribute_fhnode_attribute_default_data node_attribute_default_graphnode_attribute_default_fh attribute_named_key_ids_data attribute_named_key_ids_graphattribute_named_key_ids_fhattribute_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 rb \/var/www/ideatree/venv/lib/python3.10/site-packages/networkx/readwrite/tests/test_graphml.py setup_class
s
*
zBaseGraphML.setup_classN)__name__
__module__ __qualname__ classmethodrd rb rb rb rc 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_graphmlrD sortednodesedgesseek
parse_graphmlr: selfr HPGrb rb rc !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 ) rT r; rm rU r ro r rp rq rr rR rs rb rb rc #test_read_simple_undirected_graphmlI z3TestReadGraphML.test_read_simple_undirected_graphmlc C rx ry ) rX r; rm rY r ro r rp rq rr rV rs rb rb rc 'test_read_undirected_multigraph_graphmlT r{ z7TestReadGraphML.test_read_undirected_multigraph_graphmlc C rx ry ) r[ r; rm r\ r ro r rp rq rr rZ rs rb rb rc 4test_read_undirected_multigraph_no_multiedge_graphml_ r{ zDTestReadGraphML.test_read_undirected_multigraph_no_multiedge_graphmlc C rx ry ) r^ r; rm r_ r ro r rp rq rr r] rs rb rb rc ?test_read_undirected_multigraph_only_ids_for_multiedges_graphmlj r{ 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 rj ) rF r; rm rH r ro rn rp ziprq rr rE )rt r ru geheabrv rb rb rc test_read_attribute_graphmlu s "z+TestReadGraphML.test_read_attribute_graphmlc C s. | j }t| j}|jd |jd ksJ d S )Nr )rJ r; rm rK rG )rt r ru rb rb rc #test_node_default_attribute_graphml s z3TestReadGraphML.test_node_default_attribute_graphmlc C < d}t |d}ttjtj| ttjtj| d S )Na
r rA rB rC pytestraisesr;
NetworkXErrorrm rr rt sra rb rb rc test_directed_edge_in_undirected
z0TestReadGraphML.test_directed_edge_in_undirectedc C r )Na
r r r rb rb rc test_undirected_edge_in_directed r z0TestReadGraphML.test_undirected_edge_in_directedc C r )Na
yellow
green
blue
1.0
r r r rb rb rc test_key_raise s z TestReadGraphML.test_key_raisec C r )Na5
yellow
green
blue
r r r rb rb rc 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 ) rA rB rC r; rm rn rp rq rr rt r ra r expectedru rb rb rc test_multigraph_keys s
z$TestReadGraphML.test_multigraph_keysc C sT t }|d |d |dddtddfddtddfddtdd d
fddd td dfddd
tddfdddtddd
fg t }t || |d t j |t
d}t |j ddd|j dddsjJ |j
|j
ksrJ 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
r8 )r8 r data_key1keydata_id2r
data_key3data_id3)r r g data_key4h data_id5i data_key6data_id7r node_typeT)rl r c S ( i | ]\}}t |d d | D qS )c S r )c S s i | ] \}}t ||qS rb str).0ekeyddrb rb rc
s zWTestReadGraphML.test_preserve_multi_edge_data....r items)r nbrkey_dictrb rb rc r s zLTestReadGraphML.test_preserve_multi_edge_data...r )r nodenbr_dictrb rb rc r s zATestReadGraphML.test_preserve_multi_edge_data..)r
edge_key_typeN)r; rW r> r@ dictrA rB
write_graphmlrq rm intr rp _adjr r readrr )rt r ra ru GadjHH string_fhrb rb rc 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 Tforce_multigraph)r r
r r
r r labelr4 2r 3
shape_type rectanglez com.yworks.flowchart.terminator descriptionzdescription
line1
line2r r ) rA rB rC r; rm listrp has_edgero rq rr )rt rl ra r ru rb rb rc 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 )rA rB rC r; rm rr ro )rt r ra r ru rG rb rb rc 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 )
rA rB rC r; rm rr ro r r r ) rt goodbaduglyr ra r ru rG rb rb rc 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 rb rb )r _xrb rb rc
* zDTestReadGraphML.test_read_attributes_with_groups..Trk CustomProperty )rA rB rC r; rm ro len)rt rl ra r node_datarb rb rc 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
cudfversioni Trk r ) rA rB rC r; rm rn ro rq rr r rb rb rc test_long_attribute_type/ s
z(TestReadGraphML.test_long_attribute_typeN)re rf rg rw rz r| r} r~ r r r r r r r r r r r r r rb rb rb rc ri ; s,
- +3 &ri 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)d* Z d+S ),TestWriteGraphMLc C s t td}d S )N
lxml.etree)r rd r importorskip)r` r rb rb rc rd J s z TestWriteGraphML.setup_classc C sD zdd l }tjtjks
J W d S ty! tjtjks J Y d S w ry )
lxml.etreer; r write_graphml_lxml ImportErrorwrite_graphml_xml)rt lxmlrb rb rc test_write_interfaceO 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 Trk r= rG rA rB writerrq r; rm rn ro rp rD rt r ra ru rb rb rc 'test_write_read_simple_directed_graphmlW s
$z8TestWriteGraphML.test_write_read_simple_directed_graphmlc C s& t }| j}| }|||g d S N)r r= copy
add_graphs)rt gmlwr ru rb rb rc test_GraphMLWriter_add_graphsc 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 r4 r F) prettyprintr Trk r r rb rb rc %test_write_read_simple_no_prettyprinti 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)
named_key_idsrk c S g | ]}| qS rb r r childrb rb rc r r zTTestWriteGraphML.test_write_read_attribute_named_key_ids_graphml.. )r r7 ) attr.namer7 )r r3 r8 )r r3 )r r2 r )r r2 c s | ] \}}||kV qd S r rb )r r
r rb rb rc zSTestWriteGraphML.test_write_read_attribute_named_key_ids_graphml..c s r r rb )r r! r" rb rb rc r r )xml.etree.ElementTreer rM rA rB r rq r; rm r ro r rp rN r getrootr r allr ) rt r r ra ru xmlchildrenr default_behavior_fhnamed_key_ids_behavior_fhJrb rb rc /test_write_read_attribute_named_key_ids_graphmlv 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 Tinfer_numeric_typesrk r c S r rb r r rb rb rc r r zSTestWriteGraphML.test_write_read_attribute_numeric_type_graphml..r )z attr.typedoubler8 )r r rP rA rB r rq r; rm r ro r rp rQ r r r )rt r r ra ru r r r rb rb rc .test_write_read_attribute_numeric_type_graphml s$
z?TestWriteGraphML.test_write_read_attribute_numeric_type_graphmlc C s t }|ddg t \}}| || t |}| s"J t|j dd|j dds1J |j
|j
ks9J t | t
| 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.
r r r )r r r Tr N)r; rW r@ tempfilemkstempr rm
is_multigraphr rp r oscloseunlinkrt r fdfnameru rb rb rc 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 )NFred)namer8 r )r r )r r8 r r r r r9 r r r r edge_defaultr r )r; rS r> add_pathr? rG rA rB r rq rm r r ro r rp r rb rb rc 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
r8 r; rW r> r? rA rB r rq rm ro rp r rb rb rc 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 hellor r
r r r r r r rb rb rc %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
r8 r r r r r )r r r; rW r> int64r? rA rB r rq rm ro rp )rt npr ra ru rb rb rc %test_mixed_int_type_number_attributes s
z6TestWriteGraphML.test_mixed_int_type_number_attributesc C sx t d}|d}tddd|ifg}t \}}| || tj|t d}|j
|j
ks0J t | t
| d S )Nr$ 333333 @r8 r r r )r r float_r; rS r r
r rm r r r r r )rt r& wtr r r ru rb rb rc test_numpy_float s
z!TestWriteGraphML.test_numpy_floatc C s t }|ddg t \}}| || t |}| r"J t j|dd}| s/J t | t
| | ddd t \}}| || t |}| sWJ t j|dd}| sdJ t | t
| d S )Nr )r cr Tr r r ze-id) r; rW r@ r r
r rm r r r r r? r rb rb rc test_multigraph_to_graph s&
z)TestWriteGraphML.test_multigraph_to_graphc
C s ddl m} t }|g d dd |jD }t||d t \}}| j ||dd tj
|dd}t |}t |
|
sEJ t| | sPJ t||d t|jd d
|jd d
sfJ ||}t| } t| d ksxJ d d
| dD }
t|
t| ksJ d|} t| } t|
t|
ksJ t| t| ksJ t||d t|jd d
| jd d
sJ t| t| d S )Nr r )r r r r, r r, c S i | ]}|t |qS rb r r erb rb rc r 3 zOTestWriteGraphML.test_write_generate_edge_id_from_attribute..eidedge_id_from_attributer Trk r c S g | ]}|j d qS r
attribr edgerb rb rc r E zOTestWriteGraphML.test_write_generate_edge_id_from_attribute....//{http://graphml.graphdrawing.org/xmlns}edger )r r r; rS r@ rp set_edge_attributesr r
r generate_graphmlrm r ro r r r r findallrn valuesjoinrr r r r )
rt r r edge_attributesr r generatorru treer edge_idsrl r rb rb rc *test_write_generate_edge_id_from_attribute. s<
z;TestWriteGraphML.test_write_generate_edge_id_from_attributec
C s( ddl m} t }|g d dd |jD }t||d t \}}| j ||dd tj
|dd}t |}| s@J tj |dd }| sMJ t
| | sXJ t| | scJ td
d |jdd D t| ksxJ td
d |jddD t| ksJ ||}t| } t| dksJ dd | dD }
t|
t| ksJ d|} t| } | sJ t
| | sJ t| | sJ tdd | jdd D t| ksJ tdd | jddD t| ksJ t| t| d S )Nr r )r. r/ r0 r. c S r1 rb r r2 rb rb rc r _ r4 zZTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..r5 r6 Tr c s | ] \}}}| d V qdS r5 Ngetr uvrl rb rb rc r n zYTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..rk c s | ]\}}}|V qd S r rb r rN rO r rb rb rc r r r r c S r8 r
r9 r; rb rb rc r y r= zZTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..r> r c s rI rJ rK rM rb rb rc r rP c s rQ r rb rR rb rb rc r rS )r r r; rW r@ rp r? r r
r r@ rm r r ro r rn rB r r r rA rC rr r r r )
rt r r rD r r rE ru rF r rG graphml_datar rb rb rc 5test_multigraph_write_generate_edge_id_from_attributeZ sT
zFTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attributec C t d}|d}tddd|ifg}t \}}| || tj|t d}|j
|j
ks0J |d d d }|d d d }|t j |ddksKJ t ||jksTJ t |t
ks\J t| t| d S Nr$ r( r8 r r r gư>)abs)r r float64r; rS r r
r rm r rp approxtypefloatr r r rt r& r* r r r ru wtGwtHrb rb rc test_numpy_float64
z#TestWriteGraphML.test_numpy_float64c C rV rW )r r float32r; rS r r
r rm r rp rZ r[ r\ r r r r] rb rb rc test_numpy_float32 ra z#TestWriteGraphML.test_numpy_float32c C sr t d}| j}|d|jd d< t \}}| j||dd t |}|j
|j
ks-J t | t
| d S )Nr$ r' )r
r
r Tr )r r rP rY rp r r
r r; rm r r r r )rt r& r r r ru rb rb rc test_numpy_float64_inference s
z-TestWriteGraphML.test_numpy_float64_inferencec C s t }tdtd td }tdtd td }t}|j|d|d t \}}| || t j||d }|j |j ksAJ t
| t
| d S )
Ni( { i i i iD Radiohead)r r )
r; rS chrr r? r r
r rm r r r r )rt r name1name2r r r ru rb rb rc test_unicode_attributes s
z(TestWriteGraphML.test_unicode_attributesc C sr dd l }tdd}||}t }||jd< t }| || | d t
|}|jd |jd ks7J d S )Nr z {"a": "123"})r r ) jsonr dumpsr; rS rG rA rB r rq rm )rt rk r sar ra ru rb rb rc test_unicode_escape s
z$TestWriteGraphML.test_unicode_escapeN) re rf rg staticmethodr; r r rh rd r r r r r r
r r r! r# r' r+ r- rH rU r` rc rd rj rn rb rb rb rc r G s0
- ,:
r c @ s" e Zd ZeejZedd ZdS )TestXMLGraphMLc C s t d S r )r rd )r` rb rb rc rd s zTestXMLGraphML.setup_classN) re rf rg ro r; r r rh rd rb rb rb rc rp s
rp 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 r8 r my_list_attributeGraphML does not supportmatchN)
r r r; rS r> rA rB r TypeErrorr r ra rb rb rc 1test_exception_for_unsupported_datatype_node_attr s
"rz c C sn t d t } | jddg dd t }t jtdd t | | W d dS 1 s0w Y dS ) rq r r r8 rr rs ru rv N)
r r r; rS r? rA rB r rx r ry rb rb rc 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 )rq r rr rt ru rv N)
r r r; rS rG rA rB r rx r ry rb rb rc 2test_exception_for_unsupported_datatype_graph_attr s
"r| )rA r r r networkxr; networkx.readwrite.graphmlr networkx.utilsr r r ri r rp rz r{ r| rb rb rb rc s2 1