summaryrefslogtreecommitdiff
path: root/numpy/random/tests
Commit message (Collapse)AuthorAgeFilesLines
...
* Revert changes to mtrandKevin Sheppard2020-12-112-6/+9
| | | | This doesn't qualify for fixing under the NEP.
* BUG: Enforce high >= low on uniform number generatorsKevin Sheppard2020-12-112-0/+12
| | | | | | Check that high is weakly larger than low and raise if now closes #17905
* TST: Make test suite work in FIPS (140-2) ModeNikola Forró2020-10-293-69/+69
| | | | | | | | | Tests using MD5 algorithms fail in FIPS Mode because MD5 is not FIPS compliant. Replace MD5 with SHA256 to overcome that. Signed-off-by: Nikola Forró <nforro@redhat.com>
* conversion of strings to fstringsJakob2020-10-245-13/+12
|
* ENH: random: Add the method `permuted` to Generator. (#15121)Warren Weckesser2020-09-021-0/+50
| | | | | | | | | | | * ENH: random: Make _shuffle_raw and _shuffle_int standalone functions. * ENH: random: Add the method `permuted` to Generator. The method permuted(x, axis=None, out=None) shuffles an array. Unlike the existing shuffle method, it shuffles the slices along the given axis independently. Closes gh-5173.
* BUG: Validate output size in bin- and multinomialKevin Sheppard2020-07-142-0/+20
| | | | | | Check that size is not being broadcast closes #16833
* BLD: update cython to 0.29.21mattip2020-07-101-2/+2
|
* BUG: Ensure SeedSequence 0-padding does not collide with spawn keysRobert Kern2020-06-091-1/+27
| | | | | | | | | | | | | Fixes #16539 The implicit 0-padding that is done to small entropy inputs to make them the size of the internal pool conflicts with the spawn keys, which start with an appended 0. In order to maintain stream compatibility with unspawned `SeedSequence`s, we explicitly 0-pad short inputs out to the pool size only if the spawn key is provided, and thus would trigger the bug. This should minimize the impact on users that were not encountering the bug.
* BUG: Correct broadcasting when size is not NoneKevin Sheppard2020-06-051-0/+41
| | | | | | | Broadcastable size with inputs does not produce an error when size produces a smaller output array than the broadcast input shape. Patch checks that the output shape matches the outer shape of the broadcast fo all inputs and the size when given.
* REF: Refactor jump codeKevin Sheppard2020-05-121-5/+11
| | | | | | Refactor polynomial to be unsigned long array Remove unused code Fix md5 calculation on BE
* BUG: Correct loop order in MT19937 jumpKevin Sheppard2020-05-121-3/+43
| | | | | | Use the original loop order instead of an inverted order closes #15394
* Merge pull request #16162 from mattip/add-pxdCharles Harris2020-05-071-3/+13
|\ | | | | BUG: add missing numpy/__init__.pxd to the wheel
| * TST: add tag to __init__.pxd and test that the tag is foundmattip2020-05-061-3/+13
| |
* | BUG: Fix numpy.random.dirichlet returns NaN for small 'alpha' parameters. ↵Benjamin Trendelkamp-Schroer2020-04-291-0/+25
|/ | | | | | | | | | | | | | | (#14924) * Add stick-breaking * Add tests demonstrating slowness for beta and dirichlet generators for small alpha (and beta) values * Remove the test for beta with small `a` and `b` * Switch from standard to stick-breaking method whenever alpha.max() < 0.1 Co-authored-by: Warren Weckesser <warren.weckesser@gmail.com> Co-authored-by: Eric Wieser <wieser.eric@gmail.com> Co-authored-by: Sebastian Berg <sebastian@sipsolutions.net>
* TST: random: Skip a test if integers are 32 bit.Warren Weckesser2020-04-271-0/+2
|
* TST: random: Add more repeatability tests for random integers.Warren Weckesser2020-04-272-0/+45
| | | | | | | Add repeatability tests for when the range of the integers is `2**32` (and `2**32 +/- 1` for good measure) with broadcasting. The underlying functions called by Generator.integers and random.randint when the inputs are broadcast are different than when the inputs are scalars.
* BUG: random: Generator.integers(2**32) always returned 0.Warren Weckesser2020-04-261-0/+15
| | | | | | | | | | | | When the input to Generator.integers was 2**32, the value 2**32-1 was being passed as the `rng` argument to the 32-bit Lemire method, but that method requires `rng` be strictly less then 2**32-1. The fix was to handle 2**32-1 by calling next_uint32 directly. This also works for the legacy code without changing the stream of random integers from `randint`. Closes gh-16066.
* BUG: Alpha parameter must be 1D in `generator.dirichlet` (#15951)panpiort82020-04-182-0/+12
| | | | | Only one dimensional alpha paramter is currently supported, but higher dimensions were silently allowed and gave an incorrect results. This fixes the regression. In the future, the API could be extended to allow higher dimensional arrays for alpha. Fixes gh-15915
* Merge pull request #15876 from panpiort8/multinomial_pvals_1dMatti Picus2020-04-112-0/+12
|\ | | | | BUG: Check that `pvals` is 1D in `_generator.multinomial`.
| * Add minor tests enhancementPan Jan2020-04-062-2/+2
| |
| * Add testing and replace checkPan Jan2020-04-012-0/+12
| |
* | BUG: random: Disallow p=0 in negative_binomialWarren Weckesser2020-04-051-0/+5
| | | | | | | | | | | | | | | | | | | | | | | | Make `Generator.negative_binomial` raise a ValueError if p=0. `negative_binomial(n, p)` draws samples from the distribution of the number of failures until n successes are encountered. If p is 0, then a success is never encountered, so the probability distribution is 0 for any finite number of failures. In other words, it is not really a meaningful distribution, so we disallow p=0. Closes gh-15913.
* | Bug: Fix eigh mnd cholesky methods of numpy.random.multivariate_normalMax Balandat2020-04-041-0/+24
|/ | | | Fixes #15871
* MAINT: simplify code that assumes str/unicode and int/long are different ↵Eric Wieser2020-03-263-24/+0
| | | | | types (#15816) Cleanup from the dropping of python 2
* BUG: add missing c_distributions.pxd, enables cython use of random C-API ↵Matti Picus2020-03-162-5/+33
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | (gh-15463) xref gh-14778 As pointed out in the comment by @jamesthomasgriffin, we did not include a pxd file to expose the distribution functions documented in the random c-api. This PR adds a c_distributions.pxd file that exposes them. Squashed commits: * BUG: add missing c_distributions.pxd to enable cython use of random C-API * ENH, TST: add npyrandom library like npymath, test cython use of it * BUG: actually prefix f-string with f * MAINT: fixes from review, add _bit_generato_bit_generator.pxd * STY: fixes from review * BLD: don't use nprandom library for mtrand legacy build * TST: WindowsPath cannot be used in subprocess's list2cmdline * MAINT, API: move _bit_generator to bit_generator * DOC: add release note about moving bit_generator * DOC, MAINT: fixes from review * MAINT: redo dtype determination from review
* MAINT: cleanup unused imports; avoid redefinition of importsMike Taves2020-02-061-2/+2
| | | | | | | * Cleanup unused imports (F401) of mostly standard Python modules, or some internal but unlikely referenced modules * Where internal imports are potentially used, mark with noqa * Avoid redefinition of imports (F811)
* STY,MAINT: avoid 'multiple imports on one line' (flake8 E401)Mike Taves2020-01-281-2/+3
| | | | | | * PEP 8: "Imports should usually be on separate lines" * Where modified, sort imported modules alphabetically * Clean-up unused imports from these expanded lines
* NEP: issue deprecation warning when creating ragged array (NEP 34)Matti Picus2020-01-213-7/+10
| | | | This implements NEP 34.
* MAINT: Remove implicit inheritance from object class (#15236)Jon Dufresne2020-01-058-37/+37
| | | | | | | Inheriting from object was necessary for Python 2 compatibility to use new-style classes. In Python 3, this is unnecessary as there are no old-style classes. Dropping the object is more idiomatic Python.
* MAINT: Remove unnecessary 'from __future__ import ...' statementsJon Dufresne2020-01-032-3/+0
| | | | | As numpy is Python 3 only, these import statements are now unnecessary and don't alter runtime behavior.
* MAINT: remove dead code from reviewmattip2019-12-271-1/+0
|
* MAINT: unskip test on win32mattip2019-12-271-1/+0
|
* BUG: use tmp dir and check version for cython test (#15170)Matti Picus2019-12-261-16/+23
| | | | | | | | | | | | * BUG: use tmp dir and check version for cython test * TST, MAINT: skip on win32, fix formatting * TST: fixes from review * TST: fixes from review * TST: fixes from review
* MAINT: random: Remove a few unused imports from test files.Warren Weckesser2019-12-153-3/+0
|
* Revert "DEP: issue deprecation warning when creating ragged array (NEP 34)"revert-14794-nep-0034-implRalf Gommers2019-12-063-10/+7
|
* Merge pull request #14794 from mattip/nep-0034-implRalf Gommers2019-12-023-7/+10
|\ | | | | DEP: issue deprecation warning when creating ragged array (NEP 34)
| * MAINT: revert change to assert_array_equal, adjust testsmattip2019-10-313-3/+6
| |
| * DEP: issue deprecation warning when creating ragged array (NEP 34)mattip2019-10-313-4/+4
| |
* | Merge pull request #14954 from mattip/test-extending-cffiCharles Harris2019-11-272-2/+22
|\ \ | | | | | | TST. API: test using distributions.h via cffi
| * | TST: skip cffi tests when PYTHONOPTIMIZE>1, pycparser failsmattip2019-11-212-0/+12
| | |
| * | TST. API: test using distributions.h via cffimattip2019-11-211-2/+10
| | |
* | | TST: skip if cython is not availablemattip2019-11-221-0/+6
|/ /
* | MAINT: move numpy/random/examples -> numpy/random/_examplesmattip2019-11-191-3/+3
| |
* | API: restructure and document numpy.random C-API (#14604)Matti Picus2019-11-191-0/+32
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | * API: restructure and document numpy.random C-API * DOC: fix bad reference * API: ship, document, and start to test numpy.random C-API examples * API, DOC, TST: fix tests, refactor documentation to include snippets * BUILD: move public headers to numpy/core/include/numpy/random * TST: ignore DeprecationWarnings in setuptools and numba * DOC: document the C-API as used from Cython
* | MAINT: Remove uses of scalar aliasesEric Wieser2019-11-134-15/+14
| | | | | | | | Relates to gh-6103
* | merge latest changes on master branchRedRuM2019-11-036-29/+271
|\ \ | |/
| * BUG: random: biased samples from integers() with 8 or 16 bit dtype.Warren Weckesser2019-10-241-2/+20
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | When an 8 or 16 bit dtype was given to the integers() method of the Generator class, the resulting sample was biased. The problem was the lines of the form const uint8_t threshold = -rng_excl % rng_excl; in the implementations of Lemire's method, in the C file distributions.c. The intent was to compute (UINT8_MAX+1 - rng_excl) % rng_excl However, when the type of rng_excl has integer conversion rank lower than a C int (which is almost certainly the case for the 8 and 16 bit types), the terms in the expression -rng_excl % rng_excl are promoted to int, and the result of the calculation is always 0. The fix is to make the expression explicit, and write it as const uint8_t threshold = (UINT8_MAX - rng) % rng_excl; rng is used, because rng_excl is simply rng + 1; by using rng, we we only need the constant UINT#_MAX, without the extra +1. For consistency, I made the same change for all the data types (8, 16, 32 and 64 bit). Closes gh-14774.
| * ENH: random: Add the multivariate hypergeometric distributionWarren Weckesser2019-10-181-1/+135
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | The new method multivariate_hypergeometric(self, object colors, object nsample, size=None, method='marginals') of the class numpy.random.Generator implements the multivariate hypergeometric distribution; see https://en.wikipedia.org/wiki/Hypergeometric_distribution, specifically the section "Multivariate hypergeometric distribution". Two algorithms are implemented. The user selects which algorithm to use with the `method` parameter. The default, `method='marginals'`, is based on repeated calls of the univariate hypergeometric distribution function. The other algorithm, selected with `method='count'`, is a brute-force method that allocates an internal array of length ``sum(colors)``. It should only be used when that value is small, but it can be much faster than the "marginals" algorithm in that case. The C implementations of the two methods are in the files random_mvhg_count.c and random_mvhg_marginals.c in numpy/random/src/distributions.
| * TEST, DOC: fixes from reviewmattip2019-10-171-1/+1
| |
| * API: move bit_generator and generator to be privatemattip2019-10-112-2/+2
| |