Python Mock File Write