#!/usr/bin/env python

import os
import sys
import shutil

sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
from runtest_dirac import Filter, TestRun

test = TestRun(__file__, sys.argv)

f = Filter()
f.add(from_string = '  Magnetizabilities                     au                       SI                      cgs',
      num_lines   = 9,
      rel_tolerance   = 0.0002)


test.run(['dc_pbe'],                       ['h2o-1.xyz'],    args='--get="numerical_grid"')
os.system('../../../utils/fde-mag/dft_to_fde_grid_convert.py')

shutil.copy('numerical_grid', 'FILEEX')
shutil.copy('numerical_grid', 'GRIDOUT')
test.run(['dc_pbe_getfrozen.inp'],                       ['h2o-2.xyz'],    args='--put="numerical_grid FILEEX GRIDOUT" --get="GRIDOUT" --outcmo')
shutil.copy('GRIDOUT', 'FROZEN')

test.run(['dc_pbe_magn_g0_vw11_vw11.inp'],    ['h2o-1.xyz'], f, args='--put="numerical_grid FROZEN" --incmo')


# cleaning
# --------
os.unlink('FILEEX')
os.unlink('GRIDOUT')
os.unlink('FROZEN')
os.unlink('numerical_grid')
os.unlink('numerical_grid.ORIG')

sys.exit(test.return_code)
