Fip7ddlZddlmZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZmZddlmZddlmZdZgd Zd d gZd d gZdUdedefdZej6j9e dej6j;dde j<ddeeedfde j<ddeeedfde j<ddeeedfde j<ddededfde j>ddej@eeddfde j>ddej@eeedfde jBddeeedfd e jDddeeedfd!e jFd"dd#eed$d%fd&e jFd"dd'eed$d%fd(e jFd"dd)eed$d%fd*e jFd"dd+eed$d%fd,e jHddeeedfd-e jJdd.d/eed0fd1e jLddd2eed3fd4e jNddd5ed6fd7e jPddeed6fd8e jRddeeedfd9e jTdd:gd;ed<fd=e jVdd>d?ed@dAfdBe jXddifgdCZ-dDZ.dEZ/dFZ0dGZ1dHZ2dIZ3dJZ4dKZ5ej6j9e dLedMej6j9e dNdOZ6ej6j9e dLedMej6j9e dNdPZ7ej6j9e dLedMej6j9e dNdQZ8dRZ9dSZ:dTZ;y)VN)patch)MODEL) solutions) ASSETS_URLIS_RASPBERRYPI TORCH_VERSIONchecks) safe_download) TORCH_2_4F) r)r)r rr r)@r)r video_pathneeds_frame_countc:tj|}|js Jd|d}|jrJ|j\}}|sn4|dz }|j }|r||gn|g}||} |jrJ|j y)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 V/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/tests/test_solutions.py process_videor's  : &C <<>C6zlCC>K ,,.xxz  q ((*):%  dO ,,.KKMz>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}|jD]q\} } | jdr"t|| z || jdd<9| dk(rt|d|| <S| dk(rt|d|| <m| || <s|dk(r+t j r|di|j yt|di|||y) 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,sT5>_Y/04JF#))+ U >>' "/28e3C/DF3;;w+ , o %oo>?F3K o %oo>?F3KF3K  ##     $V $ . . 0)&)+r(ctj}gdgdgc|_|_|j t j ddddd|jvsJd|jy)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_selectionrvsj  & & (B./!BHbl 5 5r2tTJ !! !d%PQSQbQbPc#dd !r(ctj}dgdidc|_|_|j t j dddd|jrJd|j|jdk(sJd|jy)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./1A-BA*Br* 5 5q!T4H  g$STVTeTeSf"gg  ! #j'TUWUhUhTi%jj #r(ctjdtj} tjd}||y#t $rt jdYywxYw)z bhh}BHHETUVFFFu s+s1v5r9nopnq7rr5ssAA A?A::A?ctjd}|jdddid|jddddd}|Jd y ) 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_paddingrs[##6:I uajvN$$!UV@W^d$eG  ? ?? r(ctjj} |jdJd#t$r!}dt |vs Jd|Yd}~yd}~wwxYw)z=Test that update() raises ValueError for invalid config keys.{) invalid_keyz/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_argumentrsh    ) ) +Cl s #GGGu l1SV;k?ghigj=kk;ls9 A#AA#ctjdtj}tjd}||}|j Jdy)z>+ ,P.PP ,  &U!&&(//T3J7)1TT/UII   UUs D D D  Dz1VisualAISearch requires torch>=2.4 (found torch==)z1Disabled due to slow performance on Raspberry Pi.cttd|tjt |dz }|d}y)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$s<ZL <=8L''S>>7r(cddlm}tj}gdgdgddgddgddgf\|_|_|_|_|jd|jdd |_tjd tj }tj|d 5tj|d 5td5|j|}dddddddddt|sJdt!||j"dk(sJd|j"|j$dkDsJd|j$y#1swYxYw#1swYxYw#1swYxYw)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_methodrFsd?  & & (B 34 A A d 1-BHblBGRX HHQKBHHQK8B HH]"(( 3E b* +#U\\">N-O#QVWmQn#E"### fo ._2PQUV\Q]P^0__ .   ! #T'>v?R?R>S%TT #  ! !A %b)I&J`J`Ia'bb % ######s<%E0< E$EE$"E0E! E$$E- )E00E9c0tjdy)z@Test ObjectCropper init with show=True to cover display warning.Tr.N)rrBrYr(r&test_object_crop_with_show_TruerZs &r(ctjd}d|_tjdtj }t d5}t dtd5}t d 5}|j||j|j|jd d d d d d d d d y #1swYxYw#1swYxYw#1swYy xYw) 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%%40GG HH]"(( 3E | * U=sSVx-X*\eglh* u%&&($$&'') *********s=C$% C1AC 3C;C$ C CC! C$$C-)F)rs    !OO53  7j):& $N+klG   # #  t <     # #  & E   $  # #  $ud C   #  # #  (8$ G        ,,udVZ [       ,,X\ ]     $ $  t <     " "  t <        %tPZ [        $udz Z        $udz Z        %tPZ [   i))5,6\ako@pq   # #  2UD Q     # #  % >   #  * *  & 5   i))5,%Y]@^_   # #  t <   )//5,UY8Z[   ' '  %t/ R   !        wBELMEmN:ek:s@lS!* M,]^k]llm*noN+^_-`p- M,]^k]llm*noN+^_`p M,]^k]llm*noN+^_ ?`p ?c(' *r(