Add missing module keyword before test_square_pyramid() line

Thanks to MichaelAtoZ for pointing this out.