jp7ddlZddlmZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZmZddlmZddlmZdZgd Zd d gZd d gZdUdedefdZeje dejdde jddeeedfde jddeeedfde jddeeedfde jddededfde jddej eeddfde jddej eeedfde j!ddeeedfd e j"ddeeedfd!e j#d"dd#eed$d%fd&e j#d"dd'eed$d%fd(e j#d"dd)eed$d%fd*e j#d"dd+eed$d%fd,e j$ddeeedfd-e j%dd.d/eed0fd1e j&ddd2eed3fd4e j'ddd5ed6fd7e j(ddeed6fd8e j)ddeeedfd9e j*dd:gd;ed<fd=e j+dd>d?ed@dAfdBe j,ddifgdCZ-dDZ.dEZ/dFZ0dGZ1dHZ2dIZ3dJZ4dKZ5eje dLedMeje dNdOZ6eje dLedMeje dNdPZ7eje dLedMeje dNdQZ8dRZ9dSZ:dTZ;dS)VN)patch)MODEL) solutions) ASSETS_URLIS_RASPBERRYPI TORCH_VERSIONchecks) safe_download) TORCH_2_4F) r)r)r rr r)@r)r video_pathneeds_frame_countcntj|}|s Jd|d}|rU|\}}|sn;|dz }|}|r||gn|g}||} |U|dS)z^Process video with solution, feeding frames and optional frame count to the solution instance.zError reading video file rN)cv2 VideoCaptureisOpenedreadcopyrelease) solutionrrcap frame_countsuccessim0im_copyargs_s W/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/tests/test_solutions.py process_videor's : & &C <<>>CCCzCCCC>K ,,..xxzz   q ((**):I%%  HdO ,,..KKMMMMMz>Disabled for testing due to --slow test errors after YOLOE PR.)reasonzAname, solution_class, needs_frame_count, video_key, kwargs_update ObjectCounter demo_video)regionmodelshowObjectCounterVerticalvertical_videoObjectCounterwithOBBzyolo26n-obb.ptHeatmap)colormapr-r.r,HeatmapWithRegion)r3r,r-r.SpeedEstimator QueueManager LineAnalyticsTline)g@g @)analytics_typer-r.figsize PieAnalyticspie BarAnalyticsbar AreaAnalyticsarea TrackZone ObjectCropper crop_videozcropped-detections) temp_crop_dirr-r. ObjectBlurrerg{Gz?) blur_ratior-r.InstanceSegmentationyolo26n-seg.pt)r-r. VisionEye RegionCounterAIGym pose_video)r )kptsr.ParkingManager parking_video parking_model parking_areas)r-r. json_fileStreamlitInferencec |rt||nd}i}|D]\} } | dr*t|| z || dd<D| dkrt|d|| <f| dkrt|d|| <| || <|dkr/t jr|di|dSt|di|||dS) zTTest individual Ultralytics solution with video processing and parameter validation.Ntemp_rRrSrU)rrr)stritems startswithreplacer check_imshow inferencer') namesolution_classr video_key kwargs_updatetmp_pathsolution_assetsrkwargskeyvalues r& test_solutionri,saT5>G__Y//0004JF#))++  U >>' " " /28e3C/D/DF3;;w++ , , o % %ooo>>??F3KK o % %ooo>>??F3KKF3KK ###    1 N $ $V $ $ . . 0 0 0))&))+r(ctj}gdgdgc|_|_|t jddddd|jvsJd|jdS)z=Test distance calculation left click selection functionality.r r 2rlrNz+Expected track_id 1 in selected_boxes, got )rDistanceCalculationboxes track_idsmouse_event_for_distancerEVENT_LBUTTONDOWNselected_boxesdcs r&test_left_click_selectionrvsy  & ( (B.../!BHbl 5r2tTJJJ ! ! ! !#dQSQb#d#d ! ! ! ! !r(ctj}dgdidc|_|_|t jdddd|jrJd|j|jdksJd|jdS)z:Test distance calculation right click reset functionality.rrkrNz/Expected empty selected_boxes after reset, got z-Expected left_mouse_count=0 after reset, got )rrnrsleft_mouse_countrqrEVENT_RBUTTONDOWNrts r&test_right_click_resetrzs  & ( (B./1A1A1A-BA*Br* 5q!T4HHH gg"gTVTe"g"ggg ! # # #%jUWUh%j%j # # # # #r(ctjdtj} tjd}||dS#t $rt jdYdSwxYw)z>> bh}BHEEETUVVVFFFFu sss+s1vv5557rop7r7r555555555ssAA B  BB ctjd}|dddid|ddddd}| Jd d S) z>Test area chart graph update with dynamic class padding logic.r@rrcar)r count_dictplotr)rpersonNzArea chart plot returned None)rr update_graph)rplot_ims r&test_area_chart_paddingrss#6:::I uajvNNN$$!UV@W@W^d$eeG    ?     r(ctj} |dJd#t$r)}dt |vs Jd|Yd}~dSd}~wwxYw)z=Test that update() raises ValueError for invalid config keys.{) invalid_keyFz/Expected ValueError for invalid update argumentz is not a valid solution argumentz(Expected validation error message, got: N)rconfigSolutionConfigupdaterrZ)objrs r&/test_config_update_method_with_invalid_argumentrs   ) ) + +Cl s ###GGGGu lll1SVV;;;=khi=k=k;;;;;;;;;ls A A3 A..A3ctjdtj}tjd}||}|j JddS)z>+ , ,PP.PPP ,  & &U!&&((////1T71T1T////UUUUUUUUUUUUUUUI     s$(BBB/(D$$D(+D(z1VisualAISearch requires torch>=2.4 (found torch==)z1Disabled due to slow performance on Raspberry Pi.cttd|tjt |dz }|d}dS)zBTest similarity search solution with sample images and text query.z/4-imgs-similaritysearch.zip)dirz4-imgs-similaritysearchdataza dog sitting on a benchN)r rrVisualAISearchrZ)rdsearcherr%s r&test_similarity_searchr$sUZ===8LLLL'S>>>>7>>r(cddlm}tj}gdgdgddgddgddgf\|_|_|_|_|jd|jdd |_tj d tj }tj |d 5tj |d 5td5||}dddn #1swxYwYdddn #1swxYwYdddn #1swxYwYt||sJdt!||jdksJd|j|jdksJd|jdS)zLTest DistanceCalculation.process() computes distance between selected boxes.r)SolutionResults)drrr),rrrrrg?gffffff?)rr)r~r}rrextract_tracksdisplay_outputzcv2.setMouseCallbackNzExpected SolutionResults, got zExpected 2 tracks, got z Expected positive distance, got )ultralytics.solutions.solutionsrrrnrorpclssconfsrsrrrrobjectr isinstancetype total_trackspixels_distance)rruframeresults r&(test_distance_calculation_process_methodrFsc??????  & ( (B   3334 A A d 1-BHblBGRX HQKBHQK88B H]"( 3 3 3E b* + +##U\">N-O-O##QVWmQnQn##E""############################################# fo . .__0_QUV\Q]Q]0_0___ .  ! # # #%Tv?R%T%T # # #  !A % % %'b&J`'b'b % % % % %sZD0C9C" C9"C& &C9)C& *C9- D9C= =DC= DDDc0tjddS)z@Test ObjectCropper init with show=True to cover display warning.Tr.N)rrBrYr(r&test_object_crop_with_show_TruerZs &&&&&&r(c*tjd}d|_tjdtj}t d5}t dtd5}t d 5}||| | | d d d n #1swxYwYd d d n #1swxYwYd d d d S#1swxYwYd S) zVTest that display_output triggers imshow, waitKey, and destroyAllWindows when enabled.Tr)rrrrz cv2.imshowz cv2.waitKeyq) return_valuezcv2.destroyAllWindowsN) rr* env_checkrrrrordrassert_called_once)counterr mock_imshow mock_wait mock_destroys r&test_display_output_methodr_s%4000GG H]"( 3 3 3E |  * U=sSVxx-X-X-X*\eglhh* u%%%&&((($$&&&''))) ************************************************s[ D+C0;AC C0C C0 C !C0$ D0C4 4D7C4 8DD D )F)rs    !!!!!!OOOOOOOOOOOO555555333333  7 7 7j):& $N+kllG   #  t < <     #  & E E   $  #  $ud C C   #  #  (8$ G G       ,udVZ [ [      ,X\ ] ]     $  t < <     "  t < <       %tPZ [ [       $udz Z Z       $udz Z Z       %tPZ [ [   i)5,6\ako@p@pq   #  2UD Q Q     #  % > >   #  *  & 5 5   i)5,%Y]@^@^_   #  t < <   )/5,UY8Z8Z[   '  %t/ R R   !       wBEELMEEmlN:eeekkk:::sss@@@lllSSS!!!* M*n^k*n*n*nooN+^__--`_po- M*n^k*n*n*nooN+^__`_po M*n^k*n*n*nooN+^__ ? ?`_po ?ccc(''' * * * * *r(