Add test for saving notebook with .py script