The lens, the slab and the prism in code · Python lab | StemPy Lab