hh4SSKrSSKJr SSKrSSKrSSKrSSKJrJ r SSK J r SSK J r JrJrJr SSKJr SSKJr SrS rS rS rS rS rSrSr/SQrSS/rSS/rS\ R@S\\\\S.4S\ R@S\\\\S.4S\ R@S\\\\S.4S\ R@S\\S\S.4S\ RBS\\RD\\SS.4S\ RBS\\RD\\\S.4S\ RFS\\\\S.4S\ RHS\\\\S.4S \ RJS!\S"\\S#S$.4S%\ RJS!\S&\\S#S$.4S'\ RJS!\S(\\S#S$.4S)\ RJS!\S*\\S#S$.4S+\ RLS\\\\S.4S,\ RNS\\("\ S-- 5\\S..4S/\ RRS\S0\\S1.4S2\ RTS\S3\S4.4S5\ RVS\\\S4.4S6\ RXS\\\\S.4S7\ RZS\/S8Q\S9.4S:\ R\S\\("\ \- 5\\("\ \- 5S;.4S<\ R^SS04/r0SUS=\(S>\14S?jjr2\RfRi\S@SA9\RfRkSB\05SC55r6SDr7SEr8SFr9SGr:SHr;SIr\RfRi\(+SL\SM3SA9\RfRi\SNSA9SO55r?\RfRi\(+SL\SM3SA9\RfRi\SNSA9SP55r@\RfRi\(+SL\SM3SA9\RfRi\SNSA9SQ55rASRrBSSrCSTrDg)VN)patch)MODELTMP) solutions) ASSETS_URLIS_RASPBERRYPI TORCH_VERSIONchecks) safe_download) TORCH_2_4Fzsolutions_ci_demo.mp4zdecelera_landscape_min.movzsolution_ci_pose_demo.mp4zsolution_ci_parking_demo.mp4zsolution_ci_parking_areas.jsonzsolutions_ci_parking_model.ptzsolution_vertical_demo.mp4) r)r)rrr r)@r)r ObjectCounter)regionmodelshowObjectCounterVerticalObjectCounterwithOBBzyolo11n-obb.ptHeatmap)colormaprrrHeatmapWithRegion)rrrrSpeedEstimator QueueManager LineAnalyticsTline)g@g @)analytics_typerrfigsize PieAnalyticspie BarAnalyticsbar AreaAnalyticsarea TrackZone ObjectCropperzcropped-detections)crop_dirrr ObjectBlurrerg{Gz?) blur_ratiorrInstanceSegmentationyolo11n-seg.pt)rr VisionEye RegionCounterAIGym)r)kptsrParkingManager)rr json_fileStreamlitInference video_pathneeds_frame_countcl[R"U5nUR5(d SU35eSnUR5(aXUR5upVU(dO>US- nUR 5nU(aXt/OU/nU"U6n UR5(aMXUR 5 g)z^Process video with solution, feeding frames and optional frame count to the solution instance.zError reading video file rN)cv2 VideoCaptureisOpenedreadcopyrelease) solutionr:r;cap frame_countsuccessim0im_copyargs_s N/home/james-whalen/.local/lib/python3.13/site-packages/tests/test_solutions.py process_videorMs  : &C <<>>C6zlCC>K ,,..xxz  q ((*):%  dO ,,..KKMz>Disabled for testing due to --slow test errors after YOLOE PR.)reasonz6name, solution_class, needs_frame_count, video, kwargscU(a9US:wa[[SU3[S9 O[[S[3[S9 US:Xa7[[S[3[S9 [[S[ 3[S9 O7US:Xa1[ R"5(aU"S0UD6R5 gUS:Xa[OUn[U"S0UD6[[U- 5US9 g) zTTest individual Ultralytics solution with video processing and parameter validation.r/)urldirr7r9N)rDr:r;) r rrVERTICAL_VIDEOPARKING_AREAS_JSON PARKING_MODELr check_imshow inferencerMstr)namesolution_classr;videokwargss rL test_solutionr_s * *  AeW53 ?  An-=>C H ZL*<)=>CHZL-9sC % %     $V $ . . 0"&==N5E)&)sU{#+rNc[R"5n/SQ/S/sUlUlUR [ R SSSS5 SUR;deg)z=Test distance calculation left click selection functionality.rr2rbr=N)rDistanceCalculationboxes track_idsmouse_event_for_distancer>EVENT_LBUTTONDOWNselected_boxesdcs rLtest_left_click_selectionrlsU  & & (B./!BHbl 5 5r2tTJ !! !! !rNc[R"5nS/SQ0SsUlUlUR [ R SSSS5 UR(aeURS:Xdeg)z:Test distance calculation right click reset functionality.r=rarN)rrdrileft_mouse_countrgr>EVENT_RBUTTONDOWNrjs rLtest_right_click_resetrpsg  & & (B./1A-BA*Br* 5 5q!T4H    ! ## #rNc[R"S[RS9n[R"SS9nU"U5 g![ a [ R"S5 gf=f)z bhh}BHHETUVOOOu 7,A6667sA A A3A..A3c[R"SS9nURSSS0SS9 URSSSS.SS9nUceg ) z>Test area chart graph update with dynamic class padding logic.r)rr=car)r count_dictplotru)rpersonN)rr update_graph)rplot_ims rLtest_area_chart_paddingrsU##6:I uajvN$$!UV@W^d$eG   rNc[RR5nURSS9 S5e![anS[ U5;deSnAgSnAff=f)z=Test that update() raises ValueError for invalid config keys.{) invalid_keyz/Expected ValueError for invalid update argumentz is not a valid solution argumentN)rconfigSolutionConfigupdater}rZ)objrs rL/test_config_update_method_with_invalid_argumentrsV    ) ) +C< s #GGGu <1SV;;;>+ , ,, ,  &!vvx0000 'II + * ' &s C3<D3 D Dz1VisualAISearch requires torch>=2.4 (found torch==)z1Disabled due to slow performance on Raspberry Pi.c[[S3[S9 [R"[ [S- 5S9nU"S5ng)zBTest similarity search solution with sample images and text query.z/4-imgs-similaritysearch.zip)rSz4-imgs-similaritysearchdataza dog sitting on a benchN)r rrrVisualAISearchrZ)searcherrKs rLtest_similarity_searchr$s<ZL <=3G''S7P1P-QRH+,ArNcx[R"SS9n[US5(de[US5(deg)z4Test SearchApp initializes with required attributes.cpu)devicerrunN)r SearchApphasattr)apps rLtest_similarity_search_app_initr-s:   U +C 3 # ## # 3   rNc SSKJn US- n[R"USS9 [ S5HbnUR [ R"[ RRSSS5S -55nURUS US 3- 5 Md [R"[U5S 9nU"S 5nU(deg)z;Test VisualAISearch end-to-end with sample image and query.r)ImageimagesT)exist_okrru test_image_z.jpgrza red and white objectN)PILrrmakedirsrange fromarrayryr{randomrandsaverrrZ)tmp_pathr image_diriimgrrs rLtest_similarity_search_completer6s8#IKK D) 1Xoobhhryy~~c3'BS'HIJ {1#T223''S^N-OQVWmQnE"Ro-O + fo . .. .   ! ## #  ! !A %% % RoQn-O-O + +s<%E = D< D+D<#E + D9 5D<< E E  Ec,[R"SS9 g)z@Test ObjectCropper init with show=True to cover display warning.TrN)rr+rTrNrLtest_object_crop_with_show_TruerZs &rNc[R"SS9nSUl[R"S[R S9n[ S5n[ S[S5S9n[ S 5nURU5 UR5 UR5 UR5 S S S 5 S S S 5 S S S 5 g !,(df  N=f!,(df  N(=f!,(df  g =f) zVTest that display_output triggers imshow, waitKey, and destroyAllWindows when enabled.Tr)rrrurvz cv2.imshowz cv2.waitKeyq) return_valuezcv2.destroyAllWindowsN) rr env_checkryrzr{rordrassert_called_once)counterr mock_imshow mock_wait mock_destroys rLtest_display_output_methodr_s%%40GG HH]"(( 3E |  U=sSVx-X\eglh u%&&($$&'') h-X  hh-X-X  s= C' C*AC,C4C' C C C$ C'' C5)F)Er unittest.mockrr>numpyryr~testsrr ultralyticsrultralytics.utilsrrr r ultralytics.utils.downloadsr ultralytics.utils.torch_utilsr SHOW DEMO_VIDEO CROP_VIDEO POSE_VIDEO PARKING_VIDEOrVrWrUREGIONHORIZONTAL_LINE VERTICAL_LINErrCOLORMAP_PARULArrrr*r+rZr-r/r1r2r3r| Inference SOLUTIONSboolrMmarkskipif parametrizer_rlrprrrrrrrrrrrrrTrNrLr s    !OO53  $ ) ( . 5/ - 7j):&  E48   "UDA    5$?   $4dC   ((5$RVW   ((FUTXY     E48   E48   !E4JW    5$:V    5$:V   !E4JW )%%ujVV[ei:jk 334udS   eT:  && "D1 )%%ujESW:XY E48  ioouj:t2TU## cM)*Ds3QcKcGde     wB  J$N+klQS\]^m."$:7<'!( M,]^k]llm*noN+^_-`p- M,]^k]llm*noN+^_`p M,]^k]llm*noN+^_ `p &(' *rN