import numpy as np
from vmdpy import VMD as vmd

a = np.load('input.npy')
s1_ulong = np.load('ulong.npy')
u_long, u_hat_long, omega_long = vmd(a, 2000, 0, 4, 0, 1, 1e-7)
#np.save('s1_u_hat.npy', u_hat_long)
#np.save('s1_omega.npy', omega_long)

print(np.all(u_long == s1_ulong))

#print(len(omega_long))