How to do hypothesis testing in python