fix failures

chemosim-lab · Jun 25, 2024 · cb39e2b · cb39e2b
1 parent 6b49f0d
commit cb39e2b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/prolif/plotting/network.py b/prolif/plotting/network.py
@@ -35,7 +35,7 @@
 from prolif.utils import requires
 
 try:
-    from IPython.display import HTML, Javascript, display
+    from IPython.display import Javascript, display
 except ModuleNotFoundError:
     pass
 else:

diff --git a/tests/plotting/test_complex3d.py b/tests/plotting/test_complex3d.py
@@ -36,17 +36,17 @@ def plot_3d(self, fp_mols):
 
     def test_integration_display_single(self, plot_3d):
         view = plot_3d.display(display_all=False)
-        html = view._make_html()
+        html = view._view._make_html()
         assert "Hydrophobic" in html
 
     def test_integration_display_all(self, plot_3d):
         view = plot_3d.display(display_all=True)
-        html = view._make_html()
+        html = view._view._make_html()
         assert "Hydrophobic" in html
 
     def test_integration_compare(self, plot_3d):
         view = plot_3d.compare(plot_3d)
-        html = view._make_html()
+        html = view._view._make_html()
         assert "Hydrophobic" in html
 
     def test_from_fingerprint_raises_not_executed(self, ligand_mol, protein_mol):
@@ -66,5 +66,5 @@ def test_from_fingerprint_raises_not_executed(self, ligand_mol, protein_mol):
     def test_fp_plot_3d(self, fp_mols):
         fp, lig_mol, prot_mol = fp_mols
         view = fp.plot_3d(lig_mol, prot_mol, frame=0, display_all=fp.count)
-        html = view._make_html()
+        html = view._view._make_html()
         assert "Hydrophobic" in html
diff --git a/tests/plotting/test_network.py b/tests/plotting/test_network.py
@@ -96,5 +96,5 @@ def test_from_fingerprint_raises_not_executed(self, ligand_mol):
 
     def test_fp_plot_lignetwork(self, fp_mol):
         fp, lig_mol = fp_mol
-        html = fp.plot_lignetwork(lig_mol, kind="frame", frame=0, display_all=fp.count)
-        assert "<iframe" in html.data
+        view = fp.plot_lignetwork(lig_mol, kind="frame", frame=0, display_all=fp.count)
+        assert "<iframe" in view._iframe