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

Update simple.py

Browse Source 
add `sqrt` to math
This commit is contained in:
blueloveTH 2023-02-11 00:11:59 +08:00
parent f818c40398
commit b6bb543c84
3 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/simple.py
View File

@ -17,4 +17,4 @@ def test(n):
# dis(test)
# dis(is_prime)
print(test(10000))
assert test(10000) == 1229

1
src/pocketpy.h
View File

@ -580,6 +580,7 @@ void add_module_math(VM* vm){
vm->bind_func<1>(mod, "fabs", CPP_LAMBDA(vm->PyFloat(std::fabs(vm->num_to_float(args[0])))));
vm->bind_func<1>(mod, "floor", CPP_LAMBDA(vm->PyInt(std::floor(vm->num_to_float(args[0])))));
vm->bind_func<1>(mod, "ceil", CPP_LAMBDA(vm->PyInt(std::ceil(vm->num_to_float(args[0])))));
vm->bind_func<1>(mod, "sqrt", CPP_LAMBDA(vm->PyFloat(std::sqrt(vm->num_to_float(args[0])))));
}
void add_module_dis(VM* vm){

6
tests/_math.py
View File

@ -1,4 +1,4 @@
from math import log, log10, log2, sin, cos, tan, e, pi, isnan, isinf, fabs, floor, ceil
from math import log, log10, log2, sin, cos, tan, e, pi, isnan, isinf, fabs, floor, ceil, sqrt
def isclose(a, b):
return abs(a-b) < 0.000001
@ -23,4 +23,6 @@ assert isclose(fabs(-1.2), 1.2)
assert floor(1.2) == 1
assert floor(-1.2) == -2
assert ceil(1.2) == 2
assert ceil(-1.2) == -1
assert ceil(-1.2) == -1
assert isclose(sqrt(4), 2.0)