TST: fix test for lazy evaluation of 2-D wavepackets.