.hh4ddlZddlmZddlZddlZddlZddlmZm Z ddl m Z ddl m Z mZmZmZddlmZddlmZdZd Zd Zd Zd Zd ZdZdZgdZddgZddgZde j@deeeedfde j@deeeedfde j@deeeedfde j@deededfde jBdeejDeeddfde jBdeejDeeedfde jFdeeeedfde jHdeeeedfd e jJd!ed"eed#d$fd%e jJd!ed&eed#d$fd'e jJd!ed(eed#d$fd)e jJd!ed*eed#d$fd+e jLdeeeedfd,e jNdee(e d-z eed.fd/e jRded0eed1fd2e jTded3ed4fd5e jVdeeed4fd6e jXdeeeedfd7e jZdegd8ed9fd:e j\dee(e ez ee(e ez d;fde1fd?Z2ejfjied@AejfjkdBe0dCZ6dDZ7dEZ8dFZ9dGZ:dHZ;dIZejfjie dLedMAejfjiedNAdOZ?ejfjie dLedMAejfjiedNAdPZ@ejfjie dLedMAejfjiedNAdQZAdRZBdSZCdTZDy)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_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) solutionr:r;cap frame_countsuccessim0im_copyargs_s R/var/www/html/ai-service/venv/lib/python3.12/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, kwargsc|r>|dk7rttd|tnttdtt|dk(r=ttdttttdt tn0|dk(r+t jr|di|jy|dk(rtn|}t|di|tt|z |y) 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{#+rNctj}gdgdgc|_|_|j t j ddddd|jvsJy)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 !! !! !rNctj}dgdidc|_|_|j t j dddd|jrJ|jdk(sJy)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    ! ## #rNctjdtj} tjd}||y#t $rt jdYywxYw)z bhh}BHHETUVOOOu 7,A6667sAA A7A22A7ctjd}|jdddid|jddddd}|Jy ) 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   rNctjj} |jdJd#t$r}dt |vsJYd}~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 argumentN)rconfigSolutionConfigupdater}rZ)objrs rL/test_config_update_method_with_invalid_argumentrsV    ) ) +C< s #GGGu <1SV;;;>+ ,, ,  &1!vvx00001II   11s C*5C6*C36C?z1VisualAISearch requires torch>=2.4 (found torch==)z1Disabled due to slow performance on Raspberry Pi.cttdttjt tdz }|d}y)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+,ArNchtjd}t|dsJt|dsJy)z4Test SearchApp initializes with required attributes.cpu)devicerrunN)r SearchApphasattr)apps rLtest_similarity_search_app_initr-s4   U +C 3 ## # 3  rNc zddlm}|dz }tj|dt dD]`}|j t jt jjdddd z}|j|d |d z btjt| }|d }|sJy)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) 1X4oobhhryy~~c3'BS'HIJ {1#T2234''S^N-O#QVWmQn#E"### fo .. .   ! ## #  ! !A %% % ######s<%D;< D/D#D/"D;#D, (D//D8 4D;;Ec0tjdy)z@Test ObjectCropper init with show=True to cover display warning.TrN)rr+rTrNrLtest_object_crop_with_show_TruerZs &rNctjd}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)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%&&($$&'') *********s=C$% C1AC 3C;C$ C CC! C$$C-)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