1"""Tests for wannier90 module""" 2import os 3import abipy.data as abidata 4 5from abipy.core.testing import AbipyTest 6from abipy.wannier90 import Wannier90Input 7 8 9class TestWinFile(AbipyTest): 10 11 def test_win(self): 12 """Testing win API""" 13 filepath = os.path.join(abidata.dirpath, "refs", "wannier90", "example01_gaas.wout") 14 win = Wannier90Input.from_abinit_file(filepath) 15 repr(win); str(win) 16 assert win.to_string(verbose=2) 17 assert win.structure.formula == "Ga1 As1" 18 19 #if self.has_matplotlib(): 20 # assert wout.plot(show=False) 21 # assert wout.plot_centers_spread(show=False) 22 23 #if self.has_nbformat(): 24 # assert wout.write_notebook(nbpath=self.get_tmpname(text=True)) 25