Fix abiwan test

abinit · Aug 9, 2024 · f06ec9a · f06ec9a
1 parent 4aab7cb
commit f06ec9a
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/abipy/wannier90/tests/test_abiwan.py b/abipy/wannier90/tests/test_abiwan.py
@@ -42,11 +42,10 @@ def test_abiwan_without_dis(self):
 
             ebands_wan = abiwan.interpolate_ebands(line_density=5)
             assert ebands_wan.kpoints.is_path
-            plotter = abiwan.get_plotter_from_ebands(ebands_wan)
 
             if self.has_matplotlib():
                 assert abiwan.hwan.plot(show=False)
-                assert plotter.combiplot(show=False)
+                assert abiwan.plot_with_ebands(ebands_wan, show=False)
                 #assert abiwan.plot_centers_spread(show=False)
 
             if self.has_nbformat():
@@ -85,11 +84,10 @@ def test_abiwan_with_dis(self):
                     self.assert_almost_equal(ews[:n], in_eigens[spin, ik, :n])
 
             ebands_wan = abiwan.interpolate_ebands(line_density=3)
-            plotter = abiwan.get_plotter_from_ebands(ebands_wan)
 
             if self.has_matplotlib():
                 assert abiwan.hwan.plot(show=False)
-                assert plotter.combiplot(show=False)
+                assert abiwan.plot_with_ebands(ebands_wan, show=False)
 
             if self.has_nbformat():
                 assert abiwan.write_notebook(nbpath=self.get_tmpname(text=True))