Fix the strategy for the n argument of test_matrix_power()

asmeurer · asmeurer · commit 075150be1998 · 2021-10-08T19:11:53.000-06:00
diff --git a/array_api_tests/test_linalg.py b/array_api_tests/test_linalg.py
@@ -294,7 +294,7 @@ def test_matrix_norm(x, kw):
 
 @given(
     x=xps.arrays(dtype=xps.floating_dtypes(), shape=shapes),
-    n=xps.arrays(dtype=xps.floating_dtypes(), shape=shapes),
+    n=integers(),
 )
 def test_matrix_power(x, n):
     # res = _array_module.linalg.matrix_power(x, n)

Original file line number	Diff line number	Diff line change
`@@ -294,7 +294,7 @@ def test_matrix_norm(x, kw):`
`294`	`294`
`295`	`295`	`@given(`
`296`	`296`	`x=xps.arrays(dtype=xps.floating_dtypes(), shape=shapes),`
`297`		`- n=xps.arrays(dtype=xps.floating_dtypes(), shape=shapes),`
	`297`	`+ n=integers(),`
`298`	`298`	`)`
`299`	`299`	`def test_matrix_power(x, n):`
`300`	`300`	`# res = _array_module.linalg.matrix_power(x, n)`