add pytest file for the project