MAIN: Enforce ruff E501 rule

eendebakpt · eendebakpt · commit 907b73ce9bef · 2025-06-29T21:22:53.000+02:00
diff --git a/numpy/_core/tests/test_multithreading.py b/numpy/_core/tests/test_multithreading.py
@@ -272,12 +272,15 @@ def closure(b):
 def test_nonzero(dtype):
     # See: gh-28361
     #
-    # np.nonzero uses np.count_nonzero to determine the size of the output array
-    # In a second pass the indices of the non-zero elements are determined, but they can have changed
+    # np.nonzero uses np.count_nonzero to determine the size of the output.
+    # array. In a second pass the indices of the non-zero elements are
+    # determined, but they can have changed
     #
-    # This test triggers a data race which is suppressed in the TSAN CI. The test is to ensure
-    # np.nonzero does not generate a segmentation fault
+    # This test triggers a data race which is suppressed in the TSAN CI.
+    # The test is to ensure np.nonzero does not generate a segmentation fault
     x = np.random.randint(4, size=100).astype(dtype)
+    expected_warning = ('number of non-zero array elements changed'
+                        ' during function execution')
 
     def func(index):
         for _ in range(10):
@@ -287,6 +290,6 @@ def func(index):
                 try:
                     _ = np.nonzero(x)
                 except RuntimeError as ex:
-                    assert 'number of non-zero array elements changed during function execution' in str(ex)
+                    assert expected_warning in str(ex)
 
     run_threaded(func, max_workers=10, pass_count=True, outer_iterations=5)
diff --git a/ruff.toml b/ruff.toml
@@ -81,7 +81,6 @@ ignore = [
 "numpy/_core/tests/test_defchararray.py" = ["E501"]
 "numpy/_core/tests/test_einsum.py" = ["E501"]
 "numpy/_core/tests/test_multiarray.py" = ["E501"]
-"numpy/_core/tests/test_multithreading.py" = ["E501"]
 "numpy/_core/tests/test_nditer*py" = ["E501"]
 "numpy/_core/tests/test_ufunc*py" = ["E501"]
 "numpy/_core/tests/test_umath*py" = ["E501"]
