fi5 d dl mZ d dlZd dlZ ej d ZdZd Zd Z d Z
d Z d Z d
Z
G d d Z G d
d ZdS ) )sqrtNnumpy) tracemin_pcg tracemin_lulanczoslobpcgc t j d t j dd } t j t j 5 t j | d ddd dS # 1 swxY w Y dS z.Test that "tracemin_chol" raises an exception.scipy
tracemin_cholmethodN)pytest importorskipnx
barbell_graphraises
NetworkXErroralgebraic_connectivityGs /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/networkx/linalg/tests/test_algebraic_connectivity.py)test_algebraic_connectivity_tracemin_cholr s
A A r' ( ( = =
!!O<<<<= = = = = = = = = = = = = = = = = = A,,A03A0c t j d t j dd } t j t j 5 t j | d ddd dS # 1 swxY w Y dS r
)r r r r r r fiedler_vectorr s r !test_fiedler_vector_tracemin_cholr s
A A r' ( ( 5 5
!O44445 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5r c t j d t j dd } t j t j 5 t j | d ddd dS # 1 swxY w Y dS r
)r r r r r r spectral_orderingr s r $test_spectral_ordering_tracemin_cholr" s
A A r' ( ( 8 8
Q77778 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8r c t j d t j dd } t j | }t
t
j d|j d f j
}t j t j d 5 t j
j ||d d
d d
d
d
d
S # 1 swxY w Y d
S )z1Test that "tracemin_unknown" raises an exception.r r r
r )sizez Unknown linear system solver)matchFg:0yE>tracemin_unknown
normalizedtolr N)r r r r laplacian_matrixnpasarrayrandomnormalshapeTr r linalgalgebraicconnectivity_tracemin_fiedler)r LXs r $test_fiedler_vector_tracemin_unknownr7 # s
A A
A A
29##!QWQZ #99:: r r$ > r9 r
r badfecr$ r: r
r )r r r r spectral_bisectiondict enumerate
relabel_nodes)r Cmappings r test_spectral_bisectionrB / s
A A
a A III& & & & &9X&&''G
G$$A
a A WQZ, WQZ, c F t j | }|t j dd k sJ | |z }t j | }||z t j ||z d k sJ |t j ||z d k sJ d S )Nr Hz>abs)r, r2 normr approx)Alxr ynys r check_eigenvectorrO > s B
qe,,, , , , , AA B q5FM"r't444 4 4 4 4
q2v4000 0 0 0 0 0 0rC c P e Zd Zej de d Zej de d Zej de d Z d Z
ej de d Z ej de d Z d Z
ej de d Zej de d
Zej de d Zej de d Zej d
ddej fddej ff ej de d ZdS )TestAlgebraicConnectivityr c t j }t j t j t j || t j t j t j || d S )Nr )r DiGraphr r NetworkXNotImplementedr r selfr r s r
test_directedz'TestAlgebraicConnectivity.test_directedK s\ JLL
%r'@!F
b/1BAfUUUUUUrC c t j }t j t j t j || t j t j t j || | dd t j t j t j || t j t j t j || d S )Nr r )r Graphr r r r r add_edgerU s r test_null_and_singletonz1TestAlgebraicConnectivity.test_null_and_singletonS s HJJ
b&(A1VTTTT
b&(91VLLLL
1a
b&(A1VTTTT
b&(91VLLLLLLrC c t j }| t d t j | dk sJ t j t j t j || | ddd t j | dk sJ t j t j t j || d S )Nr: r r r$ weight)
r rY add_nodes_fromranger r r r r rZ rU s r test_disconnectedz+TestAlgebraicConnectivity.test_disconnected\ s HJJ q"""(++q0000
b&(91VLLLL
1a
"""(++q0000
b&(91VLLLLLLrC c t j d t j d }t j t j t j |d t j t j t j |d d S )Nr r
unknownr )r r r
path_graphr r r r rV r s r test_unrecognized_methodz2TestAlgebraicConnectivity.test_unrecognized_methodf sf G$$$ M!
b&(A1YWWWW
b&(91YOOOOOOrC c ^ t j d t j }| ddd t j | }t j |d| t j dd k sJ t j |d| }t |d| d S )
Nr r r$ r] -q=r* r r: rE rF )
r r r rY rZ r+ r rI r rO rV r r rJ rL s r test_two_nodesz(TestAlgebraicConnectivity.test_two_nodesl s G$$$ HJJ
1a
""" " "(fEEE
4J
J
J
aU6 : : :!Q"""""rC c t j d t j }| ddd | ddd | ddd dt j |d z }t j |dd
| t j d d
k sJ t j |dd
| }t |d | d S )Nr r g חA)spamr$ rm r] rh )r^ r* r rE rF )
r r r
MultiGraphrZ r+ r rI r rO rj s r test_two_nodes_multigraphz3TestAlgebraicConnectivity.test_two_nodes_multigraphx s G$$$ MOO
1ac
"""
1aa
1ab
!!! $Qv666 6(
f%
]1$
'
'
'( ( ( (
aE& I I I!Q"""""rC c x t j d t j d }t j | }dt dt d z z
}t j |dd }|t j |d k sJ t j |dd }t ||| d S ) Nr r: rh traceminri rE rF
r r r rd r+ r r rI r rO )rV r rJ sigmaacrL s r test_abbreviation_of_methodz5TestAlgebraicConnectivity.test_abbreviation_of_method s G$$$ M! " "DT!WW%%%
&qeJ
G
G
GV]5d3333333 aU: > > >!UA&&&&&rC c x t j d t j d }t j | }dt dt d z z
}t j |d| }|t j |d k sJ t j |d| }t ||| d S Nr rt r: rh ri rE rF rv rV r r rJ rw rx rL s r test_pathz#TestAlgebraicConnectivity.test_path s G$$$ M! " "DT!WW%%%
&qeF
C
C
CV]5d3333333 aU6 : : :!UA&&&&&rC c f t j d t j d }| ddg t j | }d}t j |d| }|t j |d k sJ t j |d| }t ||| d S )
Nr r
)r
r: )r r$ gʄ?rh ri rE rF )
r r r rd add_edges_fromr+ r rI r rO r| s r !test_problematic_graph_issue_2381z;TestAlgebraicConnectivity.test_problematic_graph_issue_2381 s G$$$ M! &&)*** " "
&qeF
C
C
CV]5d3333333 aU6 : : :!UA&&&&&rC c X t j d t j d }t j | }dt d z
}t j |d| }|t j |d k sJ t j |d| }t ||| d S r{
r r r cycle_graphr+ r r rI r rO r| s r
test_cyclez$TestAlgebraicConnectivity.test_cycle s G$$$ N1 " "DGG
&qeF
C
C
CV]5d3333333 aU6 : : :!UA&&&&&rC c \ t j d t j d }t j | }dt d z
}t j |d|d }|t j |d k sJ t j |d|d }t ||| d S ) Nr rt r: rh r$ )r* r seedrE rF r r| s r test_seed_argumentz,TestAlgebraicConnectivity.test_seed_argument s G$$$ N1 " "DGG
&qeF
K
K
KV]5d3333333 aU6 B B B!UA&&&&&rC )r) rw laplacian_fnFg'?TgG'91Ŵ?c v t j d t j g d } || } t j ||d| t j |d k sJ t j ||d| }t ||| d S # t j $ r}|j dvr Y d }~d S d }~ww xY w)Nr )Z)r$
)r$ ) )r$ ; )r: )r: * )r: < )r9 rp )r9 + )r9 9 )r
rt )r
, )r
: )r
)r 8 )r r )rp r )rp ) )r r )r r )rt r )rt ) )r 5 )r r )r ) )r r )r r )r )r r )r )r )r )r 1 )r )r 2 ) )r )r 6 )r )r 7 )r r )r r )r r )r r ) r )r ! )r r ) r )r " )r r )r r )r # )r r )r $ )r r )r 3 )r r )r 4 )r r ) r )r r )r r ) r )r r )r r )r / )r 0 )r - )r . )r r )r % )r & )r ' )r r )r ( )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r rh r( rE rF ))z Cholesky solver unavailable.)zLU solver unavailable.)
r r r rY r rI r rO r args) rV r) rw r r r rJ rL errs r test_buckminsterfullerenez3TestAlgebraicConnectivity.test_buckminsterfullerene s G$$$ H[
[
[
]
]
|
LOO ,jeF u$///0 0 0 0 !!
fUUUA a * * * * * x s AB B8#
B33B8N)__name__
__module__ __qualname__r mark parametrizemethodsrW r[ ra rf rk rr ry r} r r r r r+ normalized_laplacian_matrixr rC r rQ rQ J sR [ Xw//V V 0/V [ Xw//M M 0/M [ Xw//M M 0/MP P P [ Xw// # # 0/ # [ Xw// # # 0/ #' ' ' [ Xw//' ' 0/' [ Xw// ' ' 0/ ' [ Xw//' ' 0/' [ Xw//' ' 0/' [ /
&(; <
&(F G
[ Xw//l l 0/ l l lrC rQ c e Zd Zej ej ej ej fZe j
de d Z e j
de d Z
d Ze j
de d Ze j
de d Ze j
de d Ze j
de d Ze j
de d
Ze j
d d g d
g dgfdg dg dgff e j
de d ZdS )TestSpectralOrderinggraphc n | }t j t j t j | d S )N)r r r r r! rV r r s r test_nullgraphz#TestSpectralOrdering.test_nullgraph7 s- EGG
b&(>:::::::rC c Z t j d t j }| g d t j || }t
| t
| k sJ ddht
|d d t
|dd fv sJ d S )Nr )r r r )r: r9 r: r r: r9 r r$ )r r r MultiDiGraphr r! r r s r test_three_nodes_multigraphz0TestSpectralOrdering.test_three_nodes_multigraphR s G$$$ O !!"N"N"NOOO$Qv6665zzSVV####1v#eCRCj//3uQRRy>>:::::::rC c d t j d t t d }t j | t j }t j || t j
|| }||t t | fv sJ d S )Nr r r r r listr` r, r. shuffler rY add_pathr! reversedrV r pathr r s r r} z TestSpectralOrdering.test_path[ s G$$$E"II
$ HJJ
At $Qv666tHTNN 3 34444444rC c f t j d t t d }t j | t j }t j || t j
||d }||t t | fv sJ d S )Nr r r$ )r r r r s r r z'TestSpectralOrdering.test_seed_argumente s G$$$E"II
$ HJJ
At $QvA>>>tHTNN 3 34444444rC c
x t j d t j }t j |t ddd t j |t ddd t j || }t | t | k sJ t t ddd t t ddd t t ddd t t d
dd g}|d d |v sJ |d d |v sJ d S ) Nr r r r: r$ r rt r rn r r ) r r r rY r r` r! r r )rV r r r seqss r ra z&TestSpectralOrdering.test_disconnectedo s! G$$$ HJJ
AuQA'''
AuQA'''$Qv6665zzSVV#### q"a ! ! q"b!! " " q"a ! ! q"b!! " "
RaRyD QRRyD rC )r) expected_orderF)
r$ r: r r9 r
r rp r r rt )
rt r r rp r r
r9 r r: r$ T)
r$ r: r9 r r
r r rp r rt )
rt r rp r r r
r r9 r: r$ c t j d t t d }t j }t j ||d | |d |d d t j | }t j
||| }||v sJ d S ) Nr r r r] r r r$ )r) r ) r r r r` r rY r rZ r+ todenser! )rV r) r r r r rJ r s r r z TestSpectralOrdering.test_cycle s G$$$E"II HJJ
AtA&&&&
48T!WQ
/// " " * * , ,$Q:fMMM&&&&&&rC N)r r r r rY rS rq r _graphsr r r r r rf r r r r} r ra r r rC r r r 4 s xR]BODG [ Wg..A A /.A [ Wg..0 0 /.0S S S [ Xw//; ; 0/; [ Xw//; ; 0/; [ Xw//5 5 0/5 [ Xw//5 5 0/5 [ Xw//! ! 0/! [ (
3335S5S5ST U
2224R4R4RS T
[ Xw//' ' 0/ ' ' 'rC r )mathr r networkxr r r, r r r r" r7 rB rO rQ r r rC r r s
V!!
>= = =5 5 58 8 8
1 1 1g g g g g g g gT\' \' \' \' \' \' \' \' \' \'rC