@@ -91,6 +91,7 @@ def test_arange(start, stop, step, dtype):
9191 assert_array_equal (exp_array , res_array )
9292
9393
94+ @pytest .mark .parametrize ("func" , ["diag" , "diagflat" ])
9495@pytest .mark .parametrize (
9596 "k" ,
9697 [- 6 , - 5 , - 4 , - 3 , - 2 , - 1 , 0 , 1 , 2 , 3 , 4 , 5 , 6 ],
@@ -117,18 +118,22 @@ def test_arange(start, stop, step, dtype):
117118 "[[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]" ,
118119 ],
119120)
120- def test_diag_diagflat (v , k ):
121+ def test_diag_diagflat (func , v , k ):
121122 a = numpy .array (v )
122123 ia = dpnp .array (a )
123- expected = numpy . diag (a , k )
124- result = dpnp . diag (ia , k )
124+ expected = getattr ( numpy , func ) (a , k )
125+ result = getattr ( dpnp , func ) (ia , k )
125126 assert_array_equal (expected , result )
126127
127- expected = numpy .diagflat (a , k )
128- result = dpnp .diagflat (ia , k )
129- assert_array_equal (expected , result )
128+
129+ @pytest .mark .parametrize ("func" , ["diag" , "diagflat" ])
130+ def test_diag_diagflat_raise_error (func ):
131+ ia = dpnp .array ([0 , 1 , 2 , 3 , 4 ])
132+ with pytest .raises (TypeError ):
133+ getattr (dpnp , func )(ia , k = 2.0 )
130134
131135
136+ @pytest .mark .parametrize ("func" , ["diag" , "diagflat" ])
132137@pytest .mark .parametrize (
133138 "seq" ,
134139 [
@@ -144,13 +149,9 @@ def test_diag_diagflat(v, k):
144149 "[[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]" ,
145150 ],
146151)
147- def test_diag_diagflat_seq (seq ):
148- expected = numpy .diag (seq )
149- result = dpnp .diag (seq )
150- assert_array_equal (expected , result )
151-
152- expected = numpy .diagflat (seq )
153- result = dpnp .diagflat (seq )
152+ def test_diag_diagflat_seq (func , seq ):
153+ expected = getattr (numpy , func )(seq )
154+ result = getattr (dpnp , func )(seq )
154155 assert_array_equal (expected , result )
155156
156157
@@ -467,6 +468,16 @@ def test_vander(array, dtype, n, increase):
467468 assert_allclose (vander_func (numpy , a_np ), vander_func (dpnp , a_dpnp ))
468469
469470
471+ def test_vander_raise_error ():
472+ a = dpnp .array ([1 , 2 , 3 , 4 ])
473+ with pytest .raises (TypeError ):
474+ dpnp .vander (a , N = 1.0 )
475+
476+ a = dpnp .array ([[1 , 2 ], [3 , 4 ]])
477+ with pytest .raises (ValueError ):
478+ dpnp .vander (a )
479+
480+
470481@pytest .mark .parametrize (
471482 "sequence" ,
472483 [[1 , 2 , 3 , 4 ], (1 , 2 , 3 , 4 )],
0 commit comments