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]')