o Z۷i@spddlZddlZddlmZddlmZmZddlm Z ddZ ddZ Gd d d e Z e d kr6edSdS) N)njit)typeserrors)TestCasecCs|||||<dS)N)astartstopstepscalarrrQ/home/ubuntu/vllm_env/lib/python3.10/site-packages/numba/tests/test_storeslice.py setitem_slice sr cCsD|d}t||d|<d||<t||d||dd<dS)NriA)nparange)obsnPointscenterrrr usecase s"rc@seZdZddZddZdS)TestStoreSlicecCsVd}t|}|}ttjddtjft}|||t|||||dS)N ) rzeroscopyrrfloat64intprassertPreciseEqual)selfnobs_got obs_expectedcfuncrrr test_usecases   zTestStoreSlice.test_usecasec Cs2d}tjddtjtjtjtjf}t|t}tj|tjd}dddt|ddfddt|ddfdd d d d d f }|D],\}}}} tj|tjd}tj|tjd} |||||| t| |||| ||| qs  .