@@ -164,7 +164,6 @@ def test_divide_shapes(shape: tuple) -> None:
164164 """Test division operation between two arrays of the same shape"""
165165 lhs = wrapper .randu (shape , dtype .f16 )
166166 rhs = wrapper .randu (shape , dtype .f16 )
167- # Ensure no division by zero for test integrity
168167 rhs = wrapper .add (rhs , wrapper .constant (0.001 , shape , dtype .f16 ))
169168
170169 result = wrapper .div (lhs , rhs )
@@ -208,31 +207,12 @@ def test_divide_supported_dtypes(dtype_name: dtype.Dtype) -> None:
208207 shape = (5 , 5 )
209208 lhs = wrapper .randu (shape , dtype_name )
210209 rhs = wrapper .randu (shape , dtype_name )
211- # Ensure no division by zero for test integrity
212210 rhs = wrapper .add (rhs , wrapper .constant (0.001 , shape , dtype_name ))
213211
214212 result = wrapper .div (lhs , rhs )
215213 assert dtype .c_api_value_to_dtype (wrapper .get_type (result )) == dtype_name , f"Failed for dtype: { dtype_name } "
216214
217215
218- def test_divide_by0 () -> None :
219- """Test division operation for undefined error type."""
220- shape = (2 , 2 )
221- lhs = wrapper .constant (5 , shape , dtype .f16 )
222- rhs = wrapper .constant (0 , shape , dtype .f16 )
223- lhsnp = np .full (shape , 5 )
224- rhsnp = np .zeros (shape )
225- out = np .divide (lhsnp , rhsnp )
226- print (out )
227- with pytest .raises (RuntimeError ):
228- divOut = wrapper .div (lhs , rhs )
229- print (array_to_string ("" , divOut , 3 , False ))
230- wrapper .div (lhs , rhs )
231-
232- # result = wrapper.div(lhs, rhs)
233- # assert dtype.c_api_value_to_dtype(wrapper.get_type(result)) == dtype_name, f"Failed for dtype: {dtype_name}"
234-
235-
236216@pytest .mark .parametrize (
237217 "invdtypes" ,
238218 [
0 commit comments