Coverage for gpaw/test/lcao/test_basis.py: 100%

1import subprocess 

2 

3import pytest 

4 

5from gpaw.basis_data import parse_basis_filename 

6from gpaw.setup_data import search_for_file 

7 

8 

9def test_parse_basis_filename(): 

10 assert parse_basis_filename('Si.dzp.basis') == ('Si', 'dzp') 

11 assert parse_basis_filename('Si.basis') == ('Si', None) 

12 assert parse_basis_filename('Si.any.thing.basis') == ('Si', 'any.thing') 

13 

14 

15@pytest.mark.serial 

16def test_plot_basis(tmp_path): 

17 basisfile, _ = search_for_file('Ti.dzp.basis') 

18 pngfile = tmp_path / 'output.png' 

19 subprocess.check_call(['python', '-m', 'gpaw', '-T', 'plot-basis', 

20 basisfile, '--write', pngfile]) 

21 assert pngfile.is_file()