Merge pull request #666 from plotly/ensure-layout-is-saved

Ensure `plot` sends layout along.

Author: theengineear

plotly/plotly/plotly.py

@@ -220,6 +220,7 @@ def plot(figure_or_data, validate=True, **plot_options):
 
     fig = tools._replace_newline(figure)  # does not mutate figure
     data = fig.get('data', [])
+    plot_options['layout'] = fig.get('layout', {})
     response = v1.clientresp(data, **plot_options)
 
     # Check if the url needs a secret key

plotly/tests/test_core/test_plotly/test_plot.py

@@ -41,9 +41,14 @@ def test_plot_valid(self):
                     'x': [1, 2, 3],
                     'y': [2, 1, 2]
                 }
-            ]
+            ],
+            'layout': {'title': 'simple'}
         }
-        py.plot(fig, auto_open=False, filename='plot_valid')
+        url = py.plot(fig, auto_open=False, filename='plot_valid')
+        saved_fig = py.get_figure(url)
+        self.assertEqual(saved_fig['data'][0]['x'], fig['data'][0]['x'])
+        self.assertEqual(saved_fig['data'][0]['y'], fig['data'][0]['y'])
+        self.assertEqual(saved_fig['layout']['title'], fig['layout']['title'])
 
     @raises(PlotlyError)
     def test_plot_invalid(self):