Add tests for 'atomic' option in save() API