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