Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-20 03:20:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix a bug of random

Browse Source
This commit is contained in:
blueloveTH 2025-02-15 00:49:32 +08:00
parent 069e80f2f5
commit 26a073de4f
1 changed files with 6 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/modules/random.c
View File

@ -137,14 +137,7 @@ int64_t mt19937__randint(mt19937* self, int64_t a, int64_t b) {
static bool Random__new__(int argc, py_Ref argv) { static bool Random__new__(int argc, py_Ref argv) {
mt19937* ud = py_newobject(py_retval(), py_totype(argv), 0, sizeof(mt19937)); mt19937* ud = py_newobject(py_retval(), py_totype(argv), 0, sizeof(mt19937));
mt19937__ctor(ud); mt19937__ctor(ud);
if(argc == 1) return true; return true;
if(argc == 2) {
PY_CHECK_ARG_TYPE(1, tp_int);
py_i64 seed = py_toint(py_arg(1));
mt19937__seed(ud, seed);
return true;
}
return TypeError("Random(): expected 0 or 1 arguments, got %d", argc - 1);
} }
static bool Random__init__(int argc, py_Ref argv) { static bool Random__init__(int argc, py_Ref argv) {
@ -152,9 +145,11 @@ static bool Random__init__(int argc, py_Ref argv) {
// do nothing // do nothing
} else if(argc == 2) { } else if(argc == 2) {
mt19937* ud = py_touserdata(py_arg(0)); mt19937* ud = py_touserdata(py_arg(0));
PY_CHECK_ARG_TYPE(1, tp_int); if(!py_isnone(&argv[1])){
py_i64 seed = py_toint(py_arg(1)); PY_CHECK_ARG_TYPE(1, tp_int);
mt19937__seed(ud, (uint32_t)seed); py_i64 seed = py_toint(py_arg(1));
mt19937__seed(ud, (uint32_t)seed);
}
} else { } else {
return TypeError("Random(): expected 1 or 2 arguments, got %d"); return TypeError("Random(): expected 1 or 2 arguments, got %d");
} }