add test py