Change to np.allclose

[argenos]

          Are these float numbers? Maybe you want to use [`np.allclose`](https://numpy.org/doc/stable/reference/generated/numpy.allclose.html) or sth similar?

Originally posted by @minhnh in #17 (comment)