diff options
| author | mattip <matti.picus@gmail.com> | 2018-06-20 16:33:58 -0700 |
|---|---|---|
| committer | mattip <matti.picus@gmail.com> | 2018-06-21 09:36:05 -0700 |
| commit | 9013b0f60d9bc93f3c433df66757ac7589ec450b (patch) | |
| tree | 1919324138179500844728b011d43ea796929472 /numpy/random/tests/test_random.py | |
| parent | 95dada6c5a4654da6c7f3edbe409bda31ffc755b (diff) | |
| download | numpy-9013b0f60d9bc93f3c433df66757ac7589ec450b.tar.gz | |
fixes from review
Diffstat (limited to 'numpy/random/tests/test_random.py')
| -rw-r--r-- | numpy/random/tests/test_random.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/numpy/random/tests/test_random.py b/numpy/random/tests/test_random.py index a099f7be6..2e0885024 100644 --- a/numpy/random/tests/test_random.py +++ b/numpy/random/tests/test_random.py @@ -441,10 +441,11 @@ class TestRandomDist(object): assert_equal(np.random.choice(np.arange(6), s, replace=True).shape, s) # Check zero-size - assert_equal(np.random.randint(0,0,(3,0,4)).shape, (3,0,4)) - assert_equal(np.random.randint(0,-10,0).shape, (0,)) - assert_equal(np.random.choice(0,0).shape, (0,)) - assert_equal(np.random.choice([],(0,)).shape, (0,)) + assert_equal(np.random.randint(0, 0, size=(3, 0, 4)).shape, (3, 0, 4)) + assert_equal(np.random.randint(0, -10, size=0).shape, (0,)) + assert_equal(np.random.randint(10, 10, size=0).shape, (0,)) + assert_equal(np.random.choice(0, size=0).shape, (0,)) + assert_equal(np.random.choice([], size=(0,)).shape, (0,)) assert_equal(np.random.choice(['a', 'b'], size=(3, 0, 4)).shape, (3, 0, 4)) assert_raises(ValueError, np.random.choice, [], 10) |