Ewald potential breakup in python