fix

nglview/tests/test_widget.py

@@ -1025,17 +1025,17 @@ def test_write_html(mock_unset):
     display(view)
     fp = StringIO()
 
-    with patch.object(embed, 'embed_snippet') as mock_embed:
-        mock_embed.embed_snippet.return_value = 'ok'
+    with patch.object(embed, 'embed_snippet') as mock_embed_snippet:
+        mock_embed_snippet.return_value = 'ok'
         nv.write_html(fp, [view], frame_range=(0, 3))
-        mock_embed.assert_called_with([tm, cm, view])
+        mock_embed_snippet.assert_called_with([tm, cm, view])
     mock_unset.assert_called_with()
     assert len(view._ngl_coordinate_resource[0]) == 3
     assert len(view._ngl_coordinate_resource[1]) == 3
 
     # box
-    with patch.object(embed, 'embed_snippet') as mock_embed:
-        mock_embed.embed_snippet.return_value = 'ok'
+    with patch.object(embed, 'embed_snippet') as mock_embed_snippet:
+        mock_embed_snippet.return_value = 'ok'
         nv.write_html(fp, [HBox([view])], frame_range=(0, 3))
         # FIXME: assertion?
 

nglview/widget.py

@@ -7,7 +7,7 @@
 from logging import getLogger
 
 import ipywidgets as widgets
-import ipywidgets.embed
+from ipywidgets import embed
 import numpy as np
 from IPython.display import display
 from ipywidgets import (Image, Box, DOMWidget, HBox, VBox, IntSlider, Output, Play, Widget,
@@ -85,9 +85,6 @@ def write_html(fp, views, frame_range=None):
     >>> nglview.write_html('index.html', [view], frame_range=(0, 5)) # doctest: +SKIP
     """
     views = isinstance(views, DOMWidget) and [views] or views
-    embed = ipywidgets.embed
-    color = None
-    theme = None
 
     for _, v in _INIT_VIEWS.items():
         views.insert(0, v)