Fix test_bitwise_left_shift

asmeurer · asmeurer · commit abc7ebfec496 · 2024-10-21T14:31:46.000-06:00
diff --git a/array_api_tests/test_operators_and_elementwise_functions.py b/array_api_tests/test_operators_and_elementwise_functions.py
@@ -846,7 +846,7 @@ def test_bitwise_left_shift(ctx, data):
 
     binary_param_assert_dtype(ctx, left, right, res)
     binary_param_assert_shape(ctx, left, right, res)
-    nbits = res.dtype
+    nbits = dh.dtype_nbits[res.dtype]
     binary_param_assert_against_refimpl(
         ctx, left, right, res, "<<", lambda l, r: l << r if r < nbits else 0
     )

Original file line number	Diff line number	Diff line change
`@@ -846,7 +846,7 @@ def test_bitwise_left_shift(ctx, data):`
`846`	`846`
`847`	`847`	`binary_param_assert_dtype(ctx, left, right, res)`
`848`	`848`	`binary_param_assert_shape(ctx, left, right, res)`
`849`		`- nbits = res.dtype`
	`849`	`+ nbits = dh.dtype_nbits[res.dtype]`
`850`	`850`	`binary_param_assert_against_refimpl(`
`851`	`851`	`ctx, left, right, res, "<<", lambda l, r: l << r if r < nbits else 0`
`852`	`852`	`)`