U
md @ 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 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
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 rJ ^/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/networkx/readwrite/tests/test_graphml.py setup_class
s
*
zBaseGraphML.setup_classN)__name__
__module__ __qualname__ classmethodrL rJ rJ rJ rK 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 HPGrJ rJ rK !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' rU r= r rW rX r rY rZ r[ r: r\ rJ rJ rK #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 ra ) r@ r' rU rA r rW rX r rY rZ r[ r> r\ rJ rJ rK '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 ra ) rC r' rU rD r rW rX r rY rZ r[ rB r\ rJ rJ rK 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 ra ) rF r' rU rG r rW rX r rY rZ r[ rE r\ rJ rJ rK ?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 rR )
r1 r' rU r3 r rW rV rX rY ziprZ r[ r0 )r] r r^ geheabr_ rJ rJ rK test_read_attribute_graphmlu s "z+TestReadGraphML.test_read_attribute_graphmlc C s. | j }t| j}|jd |jd ks*td S )Nr )r4 r' rU r5 r2 rX )r] r r^ rJ rJ rK #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'
NetworkXErrorrU r[ r] srI rJ rJ rK test_directed_edge_in_undirected s
z0TestReadGraphML.test_directed_edge_in_undirectedc C s< d}t |d}ttjtj| ttjtj| d S )Na
r rm rq rJ rJ rK 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 rm rq rJ rJ rK test_key_raise s z TestReadGraphML.test_key_raisec C s< d}t |d}ttjtj| ttjtj| d S )Na5
yellow
green
blue
r rm rq rJ rJ rK 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' rU rV rY rX rZ r[ r] rr rI r expectedr^ rJ rJ rK 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$ r| keyZ data_key1r
Zdata_id2Z data_key3Zdata_id3)r} r
g Z data_key4h Zdata_id5i Z data_key6Zdata_id7r node_typeT)rT rx c S s( i | ] \}}t |d d | D qS )c S s( i | ] \}}t |d d | D qS )c S s i | ]\}}t ||qS rJ str).0ZekeyddrJ rJ rK
s zWTestReadGraphML.test_preserve_multi_edge_data....r items)r ZnbrZkey_dictrJ rJ rK r s zLTestReadGraphML.test_preserve_multi_edge_data...r )r nodeZnbr_dictrJ rJ rK r s
zATestReadGraphML.test_preserve_multi_edge_data..)r Z
edge_key_typeN)r' r? r) r+ r, r-
write_graphmlrZ rU intr rY rX _adjr r readr[ )r] r rI r^ ZGadjZHHZ string_fhrJ rJ rK 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' rU listrY rX Zhas_edgerW rZ r[ )r] rT rI r r^ rJ rJ rK 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' rU r[ rW rX )r] rr rI r r^ r2 rJ rJ rK 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' rU r[ rW rX rn ro rp ) r] ZgoodbadZuglyrr rI r r^ r2 rJ rJ rK 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 rJ rJ )r _xrJ rJ rK
* s zDTestReadGraphML.test_read_attributes_with_groups..TrS ZCustomProperty )r, r- r. r' rU rW lenrX )r] rT rI r Z node_datarJ rJ rK 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 TrS r )
r, r- r. r' rU rV rW rX rZ r[ ry rJ rJ rK test_long_attribute_type/ s
z(TestReadGraphML.test_long_attribute_typeN)rM rN rO r` rb rc rd re rk rl rs rt ru rv r{ r r r r r r rJ rJ rJ rK rQ ; s*
- +3 &rQ 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 rL rn importorskip)rH r rJ rJ rK rL 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 r| )r r c s s | ]\}}||kV qd S r rJ )r r r rJ rJ rK s zSTestWriteGraphML.test_write_read_attribute_named_key_ids_graphml..c s s | ]\}}||kV qd S r rJ )r r r rJ rJ rK r s )xml.etree.ElementTreer r6 r, r- r rZ r' rU r rW rX r rY r7 r getrootr r allrf ) r] r r rI r^ xmlchildrenrx Zdefault_behavior_fhZnamed_key_ids_behavior_fhJrJ rJ rK /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_typesrS r c S s g | ] }| qS rJ r r rJ rJ rK r s zSTestWriteGraphML.test_write_read_attribute_numeric_type_graphml..r| )z attr.typedoubler$ )r r r8 r, r- r rZ r' rU r rW rX r rY r9 r r r )r] r r rI r^ r r rx rJ rJ rK .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.
ri rj r| )ri rj r Trw N)r' r? r+ tempfilemkstempr rU
is_multigraphrX r rY r oscloseunlinkr] r fdfnamer^ rJ rJ rK 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 r| r r% r r r r Z edge_defaultr )r' r; r) Zadd_pathr* r2 r, r- r rZ rU r r rW rX r rY r rJ rJ rK 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 rZ rU rW rX rY r rJ rJ rK 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 rJ rJ rK %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$ r| r r r r )rn r r' r? r) Zint64r* r, r- r rZ rU rW rX rY )r] npr rI r^ rJ rJ rK %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$ r| r r )rn r Zfloat_r' r; r r r rU r r rX r r r )r] r wtr r r r^ rJ rJ rK 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 )rj cr Tr ri rj ze-id)
r' r? r+ r r r rU r rX r r r r* r rJ rJ rK 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 ri rj rj r ri r c S s i | ]}|t |qS rJ r r erJ rJ rK r 3 s zOTestWriteGraphML.test_write_generate_edge_id_from_attribute..eidZedge_id_from_attributer
TrS r| c S s g | ]}|j d qS r Zattribr edgerJ rJ rK r E s zOTestWriteGraphML.test_write_generate_edge_id_from_attribute....//{http://graphml.graphdrawing.org/xmlns}edger )r r r' r; r+ rY set_edge_attributesr r r generate_graphmlrU r rW rX r r r r findallrV valuesjoinr[ r r r )
r] r r edge_attributesr r generatorr^ treer edge_idsrT r rJ rJ rK *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 rJ r r rJ rJ rK r _ s zZTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..r r Tr c s s | ]\}}}| d V qdS r Ngetr uvrT rJ rJ rK r n s zYTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..rS c s s | ]\}}}|V qd S r rJ r r r r} rJ rJ rK r r s rw r| c S s g | ]}|j d qS r r r rJ rJ rK r y s zZTestWriteGraphML.test_multigraph_write_generate_edge_id_from_attribute..r r c s s | ]\}}}| d V qdS r r r rJ rJ rK r s c s s | ]\}}}|V qd S r rJ r rJ rJ rK r s ) r r r' r? r+ rY r r r r r rU r rX r rW r rV r r r r r r r[ r r r )
r] r r r r r r r^ r r r Z graphml_datar rJ rJ rK 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$ r| r r gư>)abs)rn r float64r' r; r r r rU r rY rX approxtypefloatr r r r] r r r r r r^ ZwtGZwtHrJ rJ rK 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 )rn r Zfloat32r' r; r r r rU r rY rX r r r r r r r rJ rJ rK 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 )rn r r8 r rY r r r r' rU r rX r r r )r] r r r r r^ rJ rJ rK 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' r; chrr r* r r r rU r rX r r r )r] r Zname1Zname2r r r r^ rJ rJ rK 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 ri z {"a": "123"}r ) jsondumpsr' r; r2 r, r- r rZ rU rX )r] r ri sar rI r^ rJ rJ rK test_unicode_escape s
z$TestWriteGraphML.test_unicode_escapeN) rM rN rO staticmethodr' r r rP rL r r r r r r r r r r r r r r r r r r
r
r rJ rJ rJ rK r G s.
- ,:
r c @ s" e Zd ZeejZedd ZdS )TestXMLGraphMLc C s t d S r )r rL )rH rJ rJ rK rL s zTestXMLGraphML.setup_classN) rM rN rO r r' r r rP rL rJ rJ rJ rK 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$ r| )my_list_attributeGraphML does not supportmatchN)
rn r r' r; r) r, r- ro TypeErrorr r rI rJ rJ rK 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 )
rn r r' r; r* r, r- ro r r r rJ rJ rK 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$ r| r r r N)
rn r r' r; r2 r, r- ro r r r rJ rJ rK 2test_exception_for_unsupported_datatype_graph_attr s
r )r, r r rn Znetworkxr' Znetworkx.readwrite.graphmlr Znetworkx.utilsr r r rQ r r r r r rJ rJ rJ rK s0 1