Vh4NdZddlmZmZddlZddlZedZGddZy)z+Unit testing for time dependent algorithms.)datetime timedeltaNi!)dayscXeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd Zy) TestCdIndexz'Unit testing for the cd index function.ctj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jd d|jd d|jd d|jd d|jd dd t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid }tj ||tj |dtdk(sJy)N r r r rr r rrrrrrtime time_delta(\?nxDiGraphadd_nodes_fromadd_edgerset_node_attributescd_index_deltaselfG node_attrss ]/home/dcms/DCMS/lib/python3.12/site-packages/networkx/algorithms/tests/test_time_dependent.pytest_common_graphzTestCdIndex.test_common_graphs JJL ;< 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 2qq!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-$1-.   q*-{{1aF3t;;;ctj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jd d|jd d|jd d|jd d|jd dd t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid }tj ||tj |dtd dk(sJy)Nr r r rr r rrrrrrdaterrrrrr)rrrrr&s r*'test_common_graph_with_given_attributesz3TestCdIndex.test_common_graph_with_given_attributes2s JJL ;< 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 2qq!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-$1-.   q*-{{1aF@DHHHr,c tj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jd d|jd d|jd d|jd d|jd dd did did did did did did did did did did did }tj||tj |dddk(sJy)Nr r r rr r rrrrrrr2FPZJrrrr r!r"r#r$r&s r*%test_common_graph_with_int_attributesz1TestCdIndex.test_common_graph_with_int_attributesUsU JJL ;< 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 2q||||||||||    q*-{{1aB/4777r,c tj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jd d|jd d|jd d|jd d|jd dd did did did did did did did did did did did }tj||tj |dddk(sJy)Nr r r rr r rrrrrrr3333334@33333>@33333sI@fffffQ@fffff&T@̬V@̌R@r3rrr8r&s r*'test_common_graph_with_float_attributesz3TestCdIndex.test_common_graph_with_float_attributesxsU JJL ;< 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 2q~~~~~~~~~~   q*-{{1aB/4777r,ctj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jd d|jd d|jd d|jd d|jd dd t dddid t dddid t dddid t dddid t dddid t dddit dddddt dddddt dddddt dddddt dddd dd }tj ||tj |dtddk(sJy)Nr r r rr r rrrrrrrrrrrr)rweightrrD)rrDg{Gz?rr&s r*test_common_graph_with_weightsz*TestCdIndex.test_common_graph_with_weightss JJL ;< 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 2qq!,-q!,-q!,-q!,-q!,-q!,- q!,: q!,: q!,: q!,:!$1-<   q*-{{1aF8DLLLr,ctj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jd d|jd d |jd d|jd dd t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t ddd id }tj ||tj |dtdk(sJy)Nr r r rr rrrrrr rrrrrrgrr&s r*test_node_with_no_predecessorsz*TestCdIndex.test_node_with_no_predecessors JJL ;< 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 2qq!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-$1-.   q*-{{1aF3s:::r,ctj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jdd|jd d|jd d |jd d|jd dd t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t ddd id }tj ||tj |dtdk(sJy)Nr rr rrr rr rrr rrrrrrrrrg?rr&s r*test_node_with_no_successorsz(TestCdIndex.test_node_with_no_successorsrNr,ctj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jd d |jd dd t d d d id t d d d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t ddd id }tj ||t jtjd5}tj|dtdddy#1swYyxYw)Nr r r rr rrrrr rrrrrrGrHrIrJrKrLzThe cd index cannot be defined.matchr rr r!r"rr#pytestraises NetworkXErrorr$r%r'r(r)ves r*test_n_equals_zerozTestCdIndex.test_n_equals_zeros JJL ;< 1a 1a 1a 1a 1a 1a 1a 1a 2qq!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-$1-.   q*- ]]   $E  1  KK1 0 1 1 1s "GGc tj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jd d |jd dd d id d id did did did did did did did did did }tj||t j tjd5}tj|dtdddy#1swYyxYw)Nr r r rr rrrrr rrr;r<r=r>r?r@rAz1Addition and comparison are not supported betweenrRr) rr r!r"r#rUrVrWr$r%rXs r*!test_time_timedelta_compatibilityz-TestCdIndex.test_time_timedelta_compatibility!sD JJL ;< 1a 1a 1a 1a 1a 1a 1a 1a 2q~~~~~~~~~~   q*- ]]   E  1 KK1 0  1 1 1s )EEctj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jdd|jd d|jd d |jd d|jd dd t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t dd d id t ddd id }tj ||t jtjd5}tj|dtdddy#1swYyxYw)Nr rr rrr rr rrr rrrrrrrr) rr r r r rrrrrz&Not all nodes have a 'time' attribute.rRrrTrXs r*test_node_with_no_timez"TestCdIndex.test_node_with_no_timeDs JJL ;< 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 2qq!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-$1-.   q*- ]]   $L  1  KK1 0 1 1 1s G00G9cztj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jd d|jd d|jd d|jd d|jd d|jd d|jd d|jd dd t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid }tj ||tj |dtdk(sJy)N rr r r r rrrrrr rr r r r rrrrrrarrrrrrrrrr&s r*test_maximally_consolidatingz(TestCdIndex.test_maximally_consolidatinghs  JJL ?@ 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 1a 2q 2q 2q 2q 2qq!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-$1-.$1-.   q*-{{1aF3r999r,c~tj}|jgd|jdd|jdd|jdd|jdd|jdd|jdd|jd d|jd d|jd d|jd dd t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid t dddid }tj ||tj |dtdk(sJy)Nr`rr r r r rrrrrrarrrrrrrrrr&s r*test_maximally_destabilizingz(TestCdIndex.test_maximally_destabilizings JJL ?@ 1a 1a 1a 1a 1a 1a 1a 1a 2q 2qq!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-q!,-$1-.$1-.   q*-{{1aF3q888r,N)__name__ __module__ __qualname____doc__r+r/r9rBrErMrPrZr\r^rcrer,r*rr sN1!rls'1(   c9c9r,