1
0
mirror of https://github.com/psychopy/psychojs.git synced 2025-05-10 18:50:54 +00:00

Fully alias numpy.random.randint

This commit is contained in:
Todd Parsons 2023-08-31 11:49:37 +01:00
parent 2343a2705c
commit 7f7b791837

View File

@ -960,12 +960,31 @@ export function turnSquareBracketsIntoArrays(input, max = 1)
* @param {number} max - one above the largest integer to be drawn
* @returns {number} a random integer in the requested range (signed)
*/
export function randint(min = 0, max)
export function randint(min, max = null, size = 1)
{
if (!Number.isInteger(size) | size < 1) {
// raise error if given an invalid size
throw {
origin: "util.random",
context: "when generating a random float",
error: "size must be a positive integer above 0",
};
}
if (size > 1) {
// if size > 1, call function multiple times with size = 1 and return an array
let values = []
for (let i = 0; i < size; i++) {
values.push(randint(min, max, 1));
}
return values
}
let lo = min;
let hi = max;
if (typeof max === "undefined")
// if no max given, go from 0 to min
if (max === null)
{
hi = lo;
lo = 0;