Modified py_test to make the histo figures of e1 and e2 properly to send to Joe Zuntz.