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 * @param {number} max - one above the largest integer to be drawn
* @returns {number} a random integer in the requested range (signed) * @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 lo = min;
let hi = max; let hi = max;
if (typeof max === "undefined") // if no max given, go from 0 to min
if (max === null)
{ {
hi = lo; hi = lo;
lo = 0; lo = 0;