MAINT: python 3.10 compatible add_note

cf data-apis#398

Author: ev-br

array_api_tests/test_array_object.py

@@ -159,7 +159,7 @@ def test_setitem(shape, dtypes, data):
                 out_val=res[idx],
             )
     except Exception as exc:
-        exc.add_note(repro_snippet)
+        ph.add_note(exc, repro_snippet)
         raise
 
 
@@ -210,7 +210,7 @@ def test_getitem_masking(shape, data):
                         out_val=out[out_idx],
                     )
     except Exception as exc:
-        exc.add_note(repro_snippet)
+        ph.add_note(exc, repro_snippet)
         raise
 
 
@@ -260,7 +260,7 @@ def test_setitem_masking(shape, data):
                     out_val=res[idx]
                 )
     except Exception as exc:
-        exc.add_note(repro_snippet)
+        ph.add_note(exc, repro_snippet)
         raise
 
 
@@ -338,7 +338,7 @@ def _test_getitem_arrays_and_ints(shape, data, idx_max_dims):
             tpl = tuple(k[idx] for k in bcast_key)
             assert out[idx] == x[tpl], f"failing at {idx = } w/ {key = }"
     except Exception as exc:
-        exc.add_note(repro_snippet)
+        ph.add_note(exc, repro_snippet)
         raise
 
 

array_api_tests/test_indexing_functions.py

@@ -70,7 +70,7 @@ def test_take(x, data):
         with pytest.raises(StopIteration):
             next(out_indices)
     except Exception as exc:
-        exc.add_note(repro_snippet)
+        ph.add_note(exc, repro_snippet)
         raise
 
 @pytest.mark.unvectorized
@@ -135,5 +135,5 @@ def test_take_along_axis(x, data):
                 for j in range(new_len):
                     assert o_1d[j] == a_1d[i_1d[j]], f'{ii=}, {kk=}, {j=}'
     except Exception as exc:
-        exc.add_note(repro_snippet)
+        ph.add_note(exc, repro_snippet)
         raise

array_api_tests/test_utility_functions.py

@@ -38,7 +38,7 @@ def test_all(x, data):
             ph.assert_scalar_equals("all", type_=scalar_type, idx=out_idx,
                                     out=result, expected=expected, kw=kw)
     except Exception as exc:
-        exc.add_note(repro_snippet)
+        ph.add_note(exc, repro_snippet)
         raise
 
 
@@ -71,7 +71,7 @@ def test_any(x, data):
             ph.assert_scalar_equals("any", type_=scalar_type, idx=out_idx,
                                     out=result, expected=expected, kw=kw)
     except Exception as exc:
-        exc.add_note(repro_snippet)
+        ph.add_note(exc, repro_snippet)
         raise
 
 
@@ -111,7 +111,7 @@ def test_diff(x, data):
                 l[n_axis] += 1
                 assert out[idx] == x[tuple(l)] - x[idx], f"diff failed with {idx = }"
     except Exception as exc:
-        exc.add_note(repro_snippet)
+        ph.add_note(exc, repro_snippet)
         raise
 
 
@@ -158,6 +158,6 @@ def test_diff_append_prepend(x, data):
         for idx in sh.ndindex(out.shape):
             assert out[idx] == out_1[idx], f"{idx = }"
     except Exception as exc:
-        exc.add_note(repro_snippet)
+        ph.add_note(exc, repro_snippet)
         raise