`L i bddlZddlZddlmZddlmZddlmZddl m Z dZ dZ dZ d Zd Zy) N)connected_components)pairwise_distances)kneighbors_graph)_fix_connected_componentsctjgddddf}t|dd}t|\}}|dkDsJt ||||}t|\}}|dk(sJy)Nrr distance n_neighborsmoder )nparrayrrrXgraphn_connected_componentslabelss d/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sklearn/utils/tests/test_graph.pytest_fix_connected_componentsr su #$QW-A QAJ ?E%9%%@"F !A %% % %a0F OE%9%%@"F !Q && &cftjgddddf}t|dd}t|\}}|dkDsJt |}t ||||d}t|\}}|dk(sJt jtd 5t ||||ddddy#1swYyxYw) Nrr rrr precomputed)metriczdoes not work with a sparsematch) rrrrrrpytestraises RuntimeError)rrrr distancess r)test_fix_connected_components_precomputedr%s #$QW-A QAJ ?E%9%%@"F !A %% %"1%I %50& E&:%%@"F !Q && & |+H I ! 50&    s B''B0ctjgddddf}t|dd}t|\}}t j t d5t||||d}dddy#1swYyxYw) Nrr rrz Unknown moderfoor)rrrrr!r" ValueErrorrrs r(test_fix_connected_components_wrong_moder*/sq #$QW-A QAJ ?E%9%%@"F z 8 ) u,f5    s A00A9ctjgddddf}t|dd}t|\}}t ||||d}tj |j dk(sJy)Nrr r r r connectivityrr()rrrrralldatars r/test_fix_connected_components_connectivity_moder0;si q$w'A QAN CE%9%%@"F % 5(&~ E 66%**/ "" "rc(tjgddddf}t|dd}tj|jdk(sJt |\}}t ||||d}tj|jdk(rJy)Nr,r rrr()rrrr.r/rrrs r+test_fix_connected_components_distance_moder2Fs q$w'A QAJ ?E 66%**/ "" "%9%%@"F % 5(&z EvvejjAo&& &&r)numpyrr!scipy.sparse.csgraphrsklearn.metrics.pairwisersklearn.neighborsrsklearn.utils.graphrrr%r*r0r2rrr9s0 57.9 ' .  # 'r