Enable callable arguments for lens()
(#581)
Add wavelength-dependent lens to the unit tests
(#581)