mirror of https://github.com/pyodide/pyodide.git
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
def test_numpy(selenium):
|
|
selenium.load_package("numpy")
|
|
selenium.run("import numpy")
|
|
x = selenium.run("numpy.zeros((32, 64))")
|
|
assert len(x) == 32
|
|
assert all(len(y) == 64 for y in x)
|
|
for y in x:
|
|
assert all(z == 0 for z in y)
|
|
|
|
|
|
def test_typed_arrays(selenium):
|
|
selenium.load_package("numpy")
|
|
selenium.run("import numpy")
|
|
for (jstype, npytype) in (
|
|
('Int8Array', 'int8'),
|
|
('Uint8Array', 'uint8'),
|
|
('Uint8ClampedArray', 'uint8'),
|
|
('Int16Array', 'int16'),
|
|
('Uint16Array', 'uint16'),
|
|
('Int32Array', 'int32'),
|
|
('Uint32Array', 'uint32'),
|
|
('Float32Array', 'float32'),
|
|
('Float64Array', 'float64')):
|
|
print(jstype, npytype)
|
|
selenium.run_js(
|
|
f'window.array = new {jstype}([1, 2, 3, 4]);\n')
|
|
assert selenium.run(
|
|
'from js import array\n'
|
|
'npyarray = numpy.asarray(array)\n'
|
|
f'npyarray.dtype.name == "{npytype}" '
|
|
'and npyarray == [1, 2, 3, 4]')
|