Add python test for new file functionality