Add test python file

Test file