From ef58852d9a9db7eae818b832c87d41f78997d4ff Mon Sep 17 00:00:00 2001 From: Hai Nguyen Date: Wed, 19 Jun 2024 22:23:41 -0400 Subject: [PATCH] more --- nglview/tests/test_widget.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/nglview/tests/test_widget.py b/nglview/tests/test_widget.py index a7ab11b5..dba952a1 100644 --- a/nglview/tests/test_widget.py +++ b/nglview/tests/test_widget.py @@ -527,7 +527,7 @@ def test_show_pymatgen(): def test_show_qcelemental(): import qcelemental as qcel - mol = qcel.models.Molecule.from_data("He 0 0 0") + mol = qcel.models.Molecule.from_data("He 0 0 0") view = nv.show_qcelemental(mol) view @@ -1026,6 +1026,7 @@ def test_write_html(mock_unset): fp = StringIO() with patch.object(embed, 'embed_snippet') as mock_embed: + mock_embed.embed_snippet.return_value = 'ok' nv.write_html(fp, [view], frame_range=(0, 3)) mock_embed.assert_called_with([tm, cm, view]) mock_unset.assert_called_with() @@ -1034,6 +1035,7 @@ def test_write_html(mock_unset): # box with patch.object(embed, 'embed_snippet') as mock_embed: + mock_embed.embed_snippet.return_value = 'ok' nv.write_html(fp, [HBox([view])], frame_range=(0, 3)) # FIXME: assertion?