1#!/usr/local/bin/python3.8
2from __future__ import print_function, unicode_literals, division, absolute_import
3import sys
4
5from abipy.data import AbinitFilesGenerator
6
7
8class MyGenerator(AbinitFilesGenerator):
9    """This class generates the output files used in the unit tests and in the examples."""
10    # Subclasses must define the following class attributes:
11    # List of pseudos (basenames) in abipy/data/pseudos
12    pseudos = ["14si.pspnc"]
13
14    # Mapping old_name --> new_name for the output files that must be preserved.
15    files_to_save = {
16        "out_DS1_DEN.nc": "si_DEN.nc",
17        "out_DS2_GSR.nc": "si_nscf_GSR.nc",
18        "out_DS2_WFK.nc": "si_nscf_WFK.nc",
19        "out_DS1_GSR.nc": "si_scf_GSR.nc",
20        "out_DS1_WFK.nc": "si_scf_WFK.nc",
21    }
22
23
24if __name__ == "__main__":
25    sys.exit(MyGenerator().run())
26