Add more NPY201 tests

crates/ruff_linter/resources/test/fixtures/numpy/NPY201.py

@@ -68,59 +68,3 @@ def func():
     np.longfloat(12+34j)
 
     np.lookfor
-
-    np.obj2sctype(int)
-
-    np.PINF
-
-    np.PZERO
-
-    np.recfromcsv
-
-    np.recfromtxt
-
-    np.round_(12.34)
-
-    np.safe_eval
-
-    np.sctype2char
-
-    np.sctypes
-
-    np.seterrobj
-
-    np.set_numeric_ops
-
-    np.set_string_function
-
-    np.singlecomplex(12+1j)
-
-    np.string_("asdf")
-
-    np.source
-
-    np.tracemalloc_domain
-
-    np.unicode_("asf")
-
-    np.who()
-
-    np.row_stack(([1,2], [3,4]))
-
-    np.alltrue([True, True])
-
-    np.anytrue([True, False])
-
-    np.cumproduct([1, 2, 3])
-
-    np.product([1, 2, 3])
-
-    np.trapz([1, 2, 3])
-
-    np.in1d([1, 2], [1, 3, 5])
-
-    np.AxisError
-
-    np.ComplexWarning
-
-    np.compare_chararrays

crates/ruff_linter/resources/test/fixtures/numpy/NPY201_2.py

@@ -0,0 +1,58 @@
+def func():
+    import numpy as np
+
+    np.obj2sctype(int)
+
+    np.PINF
+
+    np.PZERO
+
+    np.recfromcsv
+
+    np.recfromtxt
+
+    np.round_(12.34)
+
+    np.safe_eval
+
+    np.sctype2char
+
+    np.sctypes
+
+    np.seterrobj
+
+    np.set_numeric_ops
+
+    np.set_string_function
+
+    np.singlecomplex(12+1j)
+
+    np.string_("asdf")
+
+    np.source
+
+    np.tracemalloc_domain
+
+    np.unicode_("asf")
+
+    np.who()
+
+    np.row_stack(([1,2], [3,4]))
+
+    np.alltrue([True, True])
+
+    np.anytrue([True, False])
+
+    np.cumproduct([1, 2, 3])
+
+    np.product([1, 2, 3])
+
+    np.trapz([1, 2, 3])
+
+    np.in1d([1, 2], [1, 3, 5])
+
+    np.AxisError
+
+    np.ComplexWarning
+
+    np.compare_chararrays

crates/ruff_linter/resources/test/fixtures/numpy/NPY201_3.py

@@ -0,0 +1,16 @@
+def func():
+    import numpy as np
+
+    np.DTypePromotionError
+
+    np.ModuleDeprecationWarning
+
+    np.RankWarning
+
+    np.TooHardError
+
+    np.VisibleDeprecationWarning
+
+    np.chararray
+
+    np.format_parser

crates/ruff_linter/src/rules/numpy/mod.rs

@@ -16,7 +16,10 @@ mod tests {
     #[test_case(Rule::NumpyDeprecatedTypeAlias, Path::new("NPY001.py"))]
     #[test_case(Rule::NumpyLegacyRandom, Path::new("NPY002.py"))]
     #[test_case(Rule::NumpyDeprecatedFunction, Path::new("NPY003.py"))]
+    // The NPY201 tests are split into multiple files because they get fixed one by one and too many diagnostic exceed the max-iterations limit.
     #[test_case(Rule::Numpy2Deprecation, Path::new("NPY201.py"))]
+    #[test_case(Rule::Numpy2Deprecation, Path::new("NPY201_2.py"))]
+    #[test_case(Rule::Numpy2Deprecation, Path::new("NPY201_3.py"))]
     fn rules(rule_code: Rule, path: &Path) -> Result<()> {
         let snapshot = format!("{}_{}", rule_code.as_ref(), path.to_string_lossy());
         let diagnostics = test_path(