I would use `np.log(np.sum(np.exp(x.compressed())))` in the comparison, so we compare with the same as what the function is doing