Python Set Optional Parameter Angular Momentum Formula