Simplify and shorten test_fermi_net_laplacian test
Simplify and shorten test_fermi_net_laplacian test

PiperOrigin-RevId: 407303382
Change-Id: Ie5177dc14eb08d81cbaf682a6e86f6535c245e54

deepmind/ferminet