Update py
Add pytest