summaryrefslogtreecommitdiff
path: root/numpy/random
diff options
context:
space:
mode:
authorMarein Könings <marein@science.ru.nl>2017-02-28 22:23:02 +0100
committerMarein Könings <marein@science.ru.nl>2017-02-28 22:23:02 +0100
commit26768ae7066bb4417dd05de7747039edf3c116e1 (patch)
tree8489085f9be86da648f038536273a871595827d1 /numpy/random
parentee3ab365cb55cce6d0b9b6ed5cfbd8e3ede8cc66 (diff)
downloadnumpy-26768ae7066bb4417dd05de7747039edf3c116e1.tar.gz
BUG: Fix assert statements in random.choice tests
Unless I am very much mistaken these 'assert_(x,y)' calls should really be 'assert_equal(x,y)'.
Diffstat (limited to 'numpy/random')
-rw-r--r--numpy/random/tests/test_random.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/numpy/random/tests/test_random.py b/numpy/random/tests/test_random.py
index dc7d18933..e4c58e2bd 100644
--- a/numpy/random/tests/test_random.py
+++ b/numpy/random/tests/test_random.py
@@ -384,11 +384,11 @@ class TestRandomDist(TestCase):
# Check multi dimensional array
s = (2, 3)
p = [0.1, 0.1, 0.1, 0.1, 0.4, 0.2]
- assert_(np.random.choice(6, s, replace=True).shape, s)
- assert_(np.random.choice(6, s, replace=False).shape, s)
- assert_(np.random.choice(6, s, replace=True, p=p).shape, s)
- assert_(np.random.choice(6, s, replace=False, p=p).shape, s)
- assert_(np.random.choice(np.arange(6), s, replace=True).shape, s)
+ assert_equal(np.random.choice(6, s, replace=True).shape, s)
+ assert_equal(np.random.choice(6, s, replace=False).shape, s)
+ assert_equal(np.random.choice(6, s, replace=True, p=p).shape, s)
+ assert_equal(np.random.choice(6, s, replace=False, p=p).shape, s)
+ assert_equal(np.random.choice(np.arange(6), s, replace=True).shape, s)
def test_bytes(self):
np.random.seed(self.seed)