Add a bit of a test suite

numpy · Jul 6, 2018 · b38cc2b · b38cc2b
1 parent 84b230f
commit b38cc2b
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/tests/fail/random.py b/tests/fail/random.py
@@ -0,0 +1,8 @@
+import numpy as np
+
+np.random.RandomState(b"hello")  # E: incompatible type
+np.random.RandomState("")  # E: incompatible type
+
+np.random.beta(3)  # E: No overload variant
+np.random.beta(3, "hello")  # E: incompatible type
+np.random.beta(3, 4, "hello")  # E: incompatible type
diff --git a/tests/pass/random.py b/tests/pass/random.py
@@ -0,0 +1,5 @@
+import numpy as np
+
+state = np.random.RandomState()
+state = np.random.RandomState(3)
+state = np.random.RandomState([3])
diff --git a/tests/reveal/random.py b/tests/reveal/random.py
@@ -0,0 +1,4 @@
+import numpy as np
+
+np.random.beta(3, 4.0)  # E: float
+np.random.beta(3.0, 4.0, size=[3, 4])  # E: ndarray