Add more NPY002 violations (#9862)

astral-sh · Feb 7, 2024 · aa38307 · aa38307
1 parent e9ddd48
commit aa38307
Show file tree

Hide file tree

Showing 3 changed files with 292 additions and 267 deletions.
diff --git a/crates/ruff_linter/resources/test/fixtures/numpy/NPY002.py b/crates/ruff_linter/resources/test/fixtures/numpy/NPY002.py
@@ -19,8 +19,11 @@
 numpy.random.get_state()
 numpy.random.set_state()
 numpy.random.rand()
+numpy.random.ranf()
+numpy.random.sample()
 numpy.random.randn()
 numpy.random.randint()
+numpy.random.random()
 numpy.random.random_integers()
 numpy.random.random_sample()
 numpy.random.choice()
@@ -35,7 +38,6 @@
 numpy.random.f()
 numpy.random.gamma()
 numpy.random.geometric()
-numpy.random.get_state()
 numpy.random.gumbel()
 numpy.random.hypergeometric()
 numpy.random.laplace()

diff --git a/crates/ruff_linter/src/rules/numpy/rules/legacy_random.rs b/crates/ruff_linter/src/rules/numpy/rules/legacy_random.rs
@@ -80,8 +80,11 @@ pub(crate) fn legacy_random(checker: &mut Checker, expr: &Expr) {
                     "set_state" |
                     // Simple random data
                     "rand" |
+                    "ranf" |
+                    "sample" |
                     "randn" |
                     "randint" |
+                    "random" |
                     "random_integers" |
                     "random_sample" |
                     "choice" |