U
:v
f @ 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 sD d| _ t | _| jd | jjdddd | jdd | jddd d
d d d
dddg
t| j d| _
d| _ tjdd| _ ddi| j j
d< | j jddd | j jddd | j jddd | j d | j jd d d | j jddd d d! | j jdd"d#d d! | j jd"dd$d%d! | j jddd&d | j jdd d'd | j jdd d(d | j jd d d)d*d! t| j d| _d+| _tjdd| _d,d-d-d.d.d/d0| jj
d< | jd | jd" | jjdd"d d t| j d| _d1| _t | _| jjd2d3d4d5 | jjd6d7d4d5 | jjd2d6d8d9 t| j d}|| _d:| _t | _| jjdd;d< | jjd"d%d< | jjdd"d;d< | jjd"d"d d< t| j d}|| _d=| _t | _| jd | jjdddd | jddg t| j d}|| _d>| _ t | _ | j d | j jddd d | j jd"dd#d | j jdd"d$d t| j d}|| _ d?| _ t | _!| j!d | j!jddd d | j!jd"dd#d | j!jddd$d t| j d}|| _"d@| _#t | _$| j$d | j$dd | j$jd"dd#d | j$jdd"d$d t| j# d}|| _%d S )ANa
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 Zbluer Zredr 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
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 rH I/tmp/pip-unpacked-wheel-_lngutwb/networkx/readwrite/tests/test_graphml.py setup_class
s
*
zBaseGraphML.setup_classN)__name__
__module__ __qualname__ classmethodrJ rH rH rH rI 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| ks.tt| t| ksJtt|jddt|jddksnt| jd t | j
}t| t| kstt| t| kstt|jddt|jddkstd S NTdatar ) r& r% read_graphmlr- sortednodesAssertionErroredgesseek
parse_graphmlr$ selfr HPGrH rH rI !test_read_simple_directed_graphml< s $ z1TestReadGraphML.test_read_simple_directed_graphmlc C s | j }t| j}t| | s(tt| | s>t| j d t
| j }t| | sltt| | std S Nr ) r: r% rS r; r rU rV r rW rX rY r8 rZ rH rH rI #test_read_simple_undirected_graphmlI s z3TestReadGraphML.test_read_simple_undirected_graphmlc C s | j }t| j}t| | s(tt| | s>t| j d t
| j }t| | sltt| | std S r_ ) r> r% rS r? r rU rV r rW rX rY r< rZ rH rH rI 'test_read_undirected_multigraph_graphmlT s z7TestReadGraphML.test_read_undirected_multigraph_graphmlc C s | j }t| j}t| | s(tt| | s>t| j d t
| j }t| | sltt| | std S r_ ) rA r% rS rB r rU rV r rW rX rY r@ rZ rH rH rI 4test_read_undirected_multigraph_no_multiedge_graphml_ s zDTestReadGraphML.test_read_undirected_multigraph_no_multiedge_graphmlc C s | j }t| j}t| | s(tt| | s>t| j d t
| j }t| | sltt| | std S r_ ) rD r% rS rE r rU rV r rW rX rY rC rZ rH rH rI ?test_read_undirected_multigraph_only_ids_for_multiedges_graphmlj s zOTestReadGraphML.test_read_undirected_multigraph_only_ids_for_multiedges_graphmlc C s | j }t| j}t|dt|jdds2tt|jdd}t|jdd}t ||D ]\}}||ks\tq\| j
d t | j }t|dt|jddkstt|jdd}t|jdd}t ||D ]\}}||kstqd S rP )
r/ r% rS r1 r rU rT rV rW ziprX rY r. )r[ r r\ geheabr] rH rH rI test_read_attribute_graphmlu s "z+TestReadGraphML.test_read_attribute_graphmlc C s. | j }t| j}|jd |jd ks*td S )Nr )r2 r% rS r3 r0 rV )r[ r r\ rH rH rI #test_node_default_attribute_graphml s z3TestReadGraphML.test_node_default_attribute_graphmlc C s< d}t |d}ttjtj| ttjtj| d S )Na
r r* r+ r, pytestraisesr%
NetworkXErrorrS rY r[ srG rH rH rI test_directed_edge_in_undirected s
z0TestReadGraphML.test_directed_edge_in_undirectedc C s< d}t |d}ttjtj| ttjtj| d S )Na
r rk ro rH rH rI test_undirected_edge_in_directed s
z0TestReadGraphML.test_undirected_edge_in_directedc C s< d}t |d}ttjtj| ttjtj| d S )Na
yellow
green
blue
1.0
r rk ro rH rH rI test_key_raise s z TestReadGraphML.test_key_raisec C s< d}t |d}ttjtj| ttjtj| d S )Na5
yellow
green
blue
r rk ro rH rH rI test_hyperedge_raise s z$TestReadGraphML.test_hyperedge_raisec C sn d}t |d}t|}ddg}t|jdd|ks>t|d t |}t|jdd|ksjtd S )Na
r )r r r )r r r Tkeysr )
r* r+ r, r% rS rT rW rV rX rY r[ rp rG r expectedr\ rH rH rI test_multigraph_keys s
z$TestReadGraphML.test_multigraph_keysc C sL t }|d |d |dddddifddddifdddd d
fddd dd ifddd
ddifdddddd
fg t }t || |d t j|t d}t
|j ddd|j dddst |j
|j
kst dd |j
D }|d t j|ttd}||j
kst |d | }t j|ttd}||j
ksHt dS )zf
Test that data and keys of edges are preserved on consequent
write and reads
r" )r" rz keyZ data_key1r
Zdata_id2Z data_key3Zdata_id3)r{ r
g Z data_key4h Zdata_id5i Z data_key6Zdata_id7r node_typeT)rR rv c S s( i | ] \}}t |d d | D qS )c S s( i | ] \}}t |d d | D qS )c S s i | ]\}}t ||qS rH str).0ZekeyddrH rH rI
s zWTestReadGraphML.test_preserve_multi_edge_data....r items)r ZnbrZkey_dictrH rH rI r s zLTestReadGraphML.test_preserve_multi_edge_data...r )r nodeZnbr_dictrH rH rI r s
zATestReadGraphML.test_preserve_multi_edge_data..)r Z
edge_key_typeN)r% r= r' r) r* r+
write_graphmlrX rS intr rW rV _adjr r readrY )r[ r rG r\ ZGadjZHHZ string_fhrH rH rI test_preserve_multi_edge_data s:
"
z-TestReadGraphML.test_preserve_multi_edge_datac C s d}t |d}tj|dd}t| dgks8t|jdddd sLt|j d d
d ksbt|j d d
d ksxt|j d
d
dkst|j d d dkst|j d d dkst|j d
d dkst|j d
d dkst|
d t|}t| dgkst|d d d dks,t|j d d
d ksDt|j d d
d ks\t|j d
d
dkstt|j d d dkst|j d d dkst|j d
d dkst|j d
d dksttj |dd}t| dgkst|jdddd st|j d d
d ks(t|j d d
d ks@t|j d
d
dksXtt |}t| dgkszt|d d d dkst|j d d
d kst|j d d
d kst|j d
d
dkstd S )NaI
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+ r, r% rS listrW rV Zhas_edgerU rX rY )r[ rR rG r r\ rH rH rI test_yfiles_extension- sF Z
z%TestReadGraphML.test_yfiles_extensionc C s d}t |d}t|}t|}||fD ]p}|jd d sFt|jd d rXt|jd d rjt|jd d s|t|jd d rt|jd d s0tq0d S )
NaU
false
true
false
FaLsE
True
0
1
r r testr r r r
r )r* r+ r, r% rS rY rU rV )r[ rp rG r r\ r0 rH rH rI test_bool s
zTestReadGraphML.test_boolc C s d}d}d}||fD ]H}t |d}t|}t|}||fD ]}|jd d sDtqDqt |d}t tj
tj| t tj
tj| d S )Na
false
true
a
false
true
a%
false
true
r r r ) r* r+ r, r% rS rY rU rV rl rm rn ) r[ ZgoodbadZuglyrp rG r r\ r0 rH rH rI test_graphml_header_line s
z(TestReadGraphML.test_graphml_header_linec C sb d}t |d}t|}dd |jddD }t|dksDt|D ]}|d d ksHtqHd 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 rH rH )r _xrH rH rI
* s zDTestReadGraphML.test_read_attributes_with_groups..TrQ ZCustomProperty )r* r+ r, r% rS rU lenrV )r[ rR rG r Z node_datarH rH rI test_read_attributes_with_groups s
z0TestReadGraphML.test_read_attributes_with_groupsc C st d}t |d}t|}dddifg}t|jdd|ksDt|d t |}t|jdd|ksptd S ) Na
4284
r r Z cudfversioni TrQ r )
r* r+ r, r% rS rT rU rV rX rY rw rH rH rI test_long_attribute_type/ s
z(TestReadGraphML.test_long_attribute_typeN)rK rL rM r^ r` ra rb rc ri rj rq rr rs rt ry r r r r r r rH rH rH rI rO ; s*
- +3 &rO 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 rJ rl importorskip)rF r rH rH rI rJ J s z TestWriteGraphML.setup_classc C sF z dd l }tjtjkstW n$ tk
r@ tjtjks. )r
r! ) attr.namer! )r
r r" )r r )r
r rz )r r c s s | ]\}}||kV qd S r rH )r r r rH rH rI s zSTestWriteGraphML.test_write_read_attribute_named_key_ids_graphml..c s s | ]\}}||kV qd S r rH )r r r rH rH rI r s )xml.etree.ElementTreer r4 r* r+ r rX r% rS r rU rV r rW r5 r getrootr r allrd ) r[ r r rG r\ xmlchildrenrv Zdefault_behavior_fhZnamed_key_ids_behavior_fhJrH rH rI /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^t t |
|
stt t |j
dd|j
ddst | jd ||}t| }t|dkst dd |d d D }t|d kst d
|d kst d
|d ks
t d S ) Nr r TZinfer_numeric_typesrQ r c S s g | ] }| qS rH r r rH rH rI r s zSTestWriteGraphML.test_write_read_attribute_numeric_type_graphml..rz )z attr.typedoubler" )r r r6 r* r+ r rX r% rS r rU rV r rW r7 r r r )r[ r r rG r\ r r rv rH rH rI .test_write_read_attribute_numeric_type_graphml s$
z?TestWriteGraphML.test_write_read_attribute_numeric_type_graphmlc C s t }|ddg t \}}| || t |}| sDtt |j
dd|j
ddsbt|j |j ksrtt
| 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.
rg rh rz )rg rh r Tru N)r% r= r) tempfilemkstempr rS
is_multigraphrV r rW r oscloseunlinkr[ r fdfnamer\ rH rH rI test_more_multigraph_keys s
z*TestWriteGraphML.test_more_multigraph_keysc C s t jdd}|jdddd t |ddddg |jdddd d
d i|jd < d
di|jd< t }| || | d t j
|t d}t |
|
stt| | st|j|jkstd S )NZFred)namer" r )r r r rz r r# r r r r Z edge_defaultr )r% r9 r' Zadd_pathr( r0 r* r+ r rX rS r r rU rV r rW r rH rH rI test_default_attribute s
z'TestWriteGraphML.test_default_attributec C s t }|jddd |jddd |jdddd |jdddd t }| || |d t |}|j d d r~t
|j d d dkst
|j d d rt
|j d d dkst
d S ) Nr Fspecialr r r r r r r r r" r% r= r' r( r* r+ r rX rS rU rV rW r rH rH rI test_mixed_type_attributes s
z+TestWriteGraphML.test_mixed_type_attributesc C s t }|jddd |jddd |jdddd |jdddd t }| || |d t |}|j d d dkst
|j d d dkst
|j d d dkst
|j d d dkst
d S ) Nr Zhellor r r r r r r r rH rH rI %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 dkst
|j d d dkst
|jd
d dkst
|jd d dkst
d S ) Nnumpyr r r r r" rz r r r r )rl r r% r= r' Zint64r( r* r+ r rX rS rU rV rW )r[ npr rG r\ rH rH rI %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
ks`t t
| t | d S )Nr 333333 @r" rz r r )rl r Zfloat_r% r9 r r r rS r r rV r r r )r[ r wtr r r r\ rH rH rI test_numpy_float s
z!TestWriteGraphML.test_numpy_floatc C s t }|ddg t \}}| || t |}| rDtt j|dd}| s^tt
| t | | ddd t \}}| || t |}| stt j|dd}| stt
| t | d S )Nr )rh cr Tr rg rh ze-id)
r% r= r) r r r rS r rV r r r r( r rH rH rI test_multigraph_to_graph s&
z)TestWriteGraphML.test_multigraph_to_graphc
C s ddl m} t }|dddg dd |jD }t||d t \}}| j ||dd tj
|dd }t |}t |
|
stt| | stt||d
t|jd d |jd d st||}t| } t| d
kstdd | dD }
t|
t| ks$td|} t| } t|
t|
ksVtt| t| ksttt||d
t|jd d | jd d stt| t| d S )Nr r rg rh rh r rg r c S s i | ]}|t |qS rH r r erH rH rI r 3 s zOTestWriteGraphML.test_write_generate_edge_id_from_attribute..eidZedge_id_from_attributer
TrQ rz c S s g | ]}|j d qS r attribr ZedgerH rH rI r E s zOTestWriteGraphML.test_write_generate_edge_id_from_attribute....//{http://graphml.graphdrawing.org/xmlns}edger )r r r% r9 r) rW set_edge_attributesr r r generate_graphmlrS r rU rV r r r r findallrT valuesjoinrY r r r )
r[ r r edge_attributesr r generatorr\ treer edge_idsrR r rH rH rI *test_write_generate_edge_id_from_attribute. s<
z;TestWriteGraphML.test_write_generate_edge_id_from_attributec
C s: ddl m} t }|ddddg dd |jD }t||d t \}}| j ||dd tj
|dd }t |}| st
tj |d
d }| st
t| | st
t| | st
td d
|jd
dD t| kst
tdd
|jd
dD t| ks t
||}t| } t| dksFt
dd | dD }
t|
t| ksxt
d|} t| } | st
t| | st
t| | st
tdd
| jd
dD t| kst
tdd
| jd
dD t| ks"t
t| t| d S )Nr r r r r c S s i | ]}|t |qS rH r r rH rH rI r _ s zZTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..r r Tr c s s | ]\}}}| d V qdS r Ngetr uvrR rH rH rI r n s zYTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..rQ c s s | ]\}}}|V qd S r rH r r r r{ rH rH rI r r s ru rz c S s g | ]}|j d qS r r r rH rH rI r y s zZTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..r r c s s | ]\}}}| d V qdS r r r rH rH rI r s c s s | ]\}}}|V qd S r rH r rH rH rI r s ) r r r% r= r) rW r r r r r rS r rV r rU r rT r r r r r r rY r r r )
r[ r r r r r r r\ r r r Z graphml_datar rH rH rI 5test_multigraph_write_generate_edge_id_from_attributeZ sT
zFTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attributec C s t d}|d}tddd|ifg}t \}}| || tj|t d}|j
|j
ks`t |d d d }|d d d }|t j |ddkst t
||jkst t
|tkst t| t| d S Nr r r" rz r r gư>)abs)rl r float64r% r9 r r r rS r rW rV approxtypefloatr r r r[ r r r r r r\ ZwtGZwtHrH rH rI test_numpy_float64 s
z#TestWriteGraphML.test_numpy_float64c C s t d}|d}tddd|ifg}t \}}| || tj|t d}|j
|j
ks`t |d d d }|d d d }|t j |ddkst t
||jkst t
|tkst t| t| d S r )rl r Zfloat32r% r9 r r r rS r rW rV r r r r r r r rH rH rI test_numpy_float32 s
z#TestWriteGraphML.test_numpy_float32c C sr t d}| j}|d|jd d< t \}}| j||dd t |}|j
|j
ksZt t
| t | d S )Nr r )r r r Tr )rl r r6 r rW r r r r% rS r rV r r r )r[ r r r r r\ rH rH rI 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 kst
t | t
| d S )
Ni( { i i i iD Z Radiohead)r r )r% r9 chrr r( r r r rS r rV r r r )r[ r Zname1Zname2r r r r\ rH rH rI test_unicode_attributes s
z(TestWriteGraphML.test_unicode_attributesc C sp dd l }ddi}||}t }||jd< t }| || |d t |}|jd |jd kslt
d S )Nr rg z {"a": "123"}r ) jsondumpsr% r9 r0 r* r+ r rX rS rV )r[ r
rg sar rG r\ rH rH rI test_unicode_escape s
z$TestWriteGraphML.test_unicode_escapeN) rK rL rM staticmethodr% r r rN rJ r r r r r r r r r r r r r r r r r r r r rH rH rH rI r G s.
- ,:
r c @ s" e Zd ZeejZedd ZdS )TestXMLGraphMLc C s t d S r )r rJ )rF rH rH rI rJ s zTestXMLGraphML.setup_classN) rK rL rM r r% r r rN rJ rH rH rH rI r s
r c C sX t d t } | jddddgd t }t jtdd t | | W 5 Q R X 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" rz )my_list_attributeGraphML does not supportmatchN)
rl r r% r9 r' r* r+ rm TypeErrorr r rG rH rH rI 1test_exception_for_unsupported_datatype_node_attr s
r c C sZ t d t } | jdddddgd t }t jtdd t | | W 5 Q R X dS r )
rl r r% r9 r( r* r+ rm r r r rH rH rI 1test_exception_for_unsupported_datatype_edge_attr s
r c C sT t d t } dddg| jd< t }t jtdd t | | W 5 Q R X dS ) r r r r" rz r r r N)
rl r r% r9 r0 r* r+ rm r r r rH rH rI 2test_exception_for_unsupported_datatype_graph_attr s
r )r* r r rl Znetworkxr% Znetworkx.readwrite.graphmlr Znetworkx.utilsr r r rO r r r r r rH rH rH rI s0 1