pytest testing

added pytest testing