Support auto-resize in offline.plot with output_type='div' and include_plotlyjs=False

Author: jonmmease

plotly/offline/offline.py

@@ -541,7 +541,12 @@ def plot(figure_or_data, show_link=True, link_text='Export to plot.ly',
                 '</div>',
             ])
         else:
-            return plot_html
+            return ''.join([
+                '<div>',
+                plot_html,
+                resize_script,
+                '</div>',
+            ])
 
 
 def plot_mpl(mpl_fig, resize=False, strip_style=False,

plotly/tests/test_core/test_offline/test_offline.py

@@ -21,6 +21,13 @@
     )
 }
 
+
+resize_code_strings = [
+            'window.addEventListener("resize", ',
+            'Plotly.Plots.resize('
+        ]
+
+
 PLOTLYJS = plotly.offline.offline.get_plotlyjs()
 
 
@@ -77,10 +84,7 @@ def test_div_output(self):
         self.assertTrue(html.startswith('<div>') and html.endswith('</div>'))
 
     def test_autoresizing(self):
-        resize_code_strings = [
-            'window.addEventListener("resize", ',
-            'Plotly.Plots.resize('
-        ]
+
         # If width or height wasn't specified, then we add a window resizer
         html = self._read_html(plotly.offline.plot(fig, auto_open=False))
         for resize_code_string in resize_code_strings:
@@ -96,6 +100,28 @@ def test_autoresizing(self):
         for resize_code_string in resize_code_strings:
             self.assertNotIn(resize_code_string, html)
 
+    def test_autoresizing_div(self):
+
+        # If width or height wasn't specified, then we add a window resizer
+        for include_plotlyjs in [True, False]:
+            html = plotly.offline.plot(fig,
+                                       output_type='div',
+                                       include_plotlyjs=include_plotlyjs)
+
+            for resize_code_string in resize_code_strings:
+                self.assertIn(resize_code_string, html)
+
+        # If width or height was specified, then we don't resize
+        html = plotly.offline.plot({
+            'data': fig['data'],
+            'layout': {
+                'width': 500, 'height': 500
+            }
+        }, output_type='div')
+
+        for resize_code_string in resize_code_strings:
+            self.assertNotIn(resize_code_string, html)
+
     def test_config(self):
         config = dict(linkText='Plotly rocks!',
                       editable=True)