Jumpy bug

jumpy_bug.py

import jumpy
import numpy as np
jumpy.init()


def numpy(jp_array):
    # convert back to numpy array
    array = jp_array.array  # INDArray instance
    get = array.getDouble
    shape = array.shape()
    length = array.length()
    scalars = [get(i) for i in range(length)]
    return np.array(scalars).reshape(shape)


np_small = np.zeros((32,))
jp_small = jumpy.from_np(np_small)
np_small = numpy(jp_small)
assert np_small.max() == 0

np_big = np.zeros((3200,))
jp_big = jumpy.from_np(np_big)
np_big = numpy(jp_big)
assert np_big.max() == 0, np_big.max()