Added a new test python file for the class