Skip to content

Commit e4461b3

Browse files
committed
Improve parametrize names for scalar promotion tests
1 parent 3f1060f commit e4461b3

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

array_api_tests/test_type_promotion.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -374,10 +374,12 @@ def test_operator_inplace_two_arg_promoted_promotion(binary_op_name, dtypes, two
374374

375375
assert res.dtype == res_dtype, f"{dtype1} {binary_op}= {dtype2} promoted to {res.dtype}, should have promoted to {res_dtype} (shape={shape1, shape2})"
376376

377-
scalar_promotion_parametrize_inputs = [(binary_op_name, dtype, scalar_type)
378-
for binary_op_name in sorted(set(binary_operators) - {'__matmul__'})
379-
for dtype in input_types[elementwise_function_input_types[operators_to_functions[binary_op_name]]]
380-
for scalar_type in dtypes_to_scalars[dtype]]
377+
scalar_promotion_parametrize_inputs = [
378+
pytest.param(binary_op_name, dtype, scalar_type, id=f"{binary_op_name}-{dtype}-{scalar_type.__name__}")
379+
for binary_op_name in sorted(set(binary_operators) - {'__matmul__'})
380+
for dtype in input_types[elementwise_function_input_types[operators_to_functions[binary_op_name]]]
381+
for scalar_type in dtypes_to_scalars[dtype]
382+
]
381383

382384
@pytest.mark.parametrize('binary_op_name,dtype,scalar_type',
383385
scalar_promotion_parametrize_inputs)

0 commit comments

Comments
 (0)