diff options
author | Marein Könings <marein@science.ru.nl> | 2017-02-28 22:23:02 +0100 |
---|---|---|
committer | Marein Könings <marein@science.ru.nl> | 2017-02-28 22:23:02 +0100 |
commit | 26768ae7066bb4417dd05de7747039edf3c116e1 (patch) | |
tree | 8489085f9be86da648f038536273a871595827d1 /numpy/random | |
parent | ee3ab365cb55cce6d0b9b6ed5cfbd8e3ede8cc66 (diff) | |
download | numpy-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.py | 10 |
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) |