import numpy as np

from ase.calculators.eam import EAM

from ase.test.eam_pot import Pt_u3
from ase.build import fcc111
import os

# test to read EAM potential from *.eam file (aka funcfl format) - for one element

pot_fn = 'Pt_u3.eam'
f = open(pot_fn,'w')
f.write(Pt_u3)
f.close()

eam = EAM(potential='Pt_u3.eam', elements=['Pt'])
slab = fcc111('Pt', size=(4, 4, 2), vacuum=10.0)
slab.set_calculator(eam)

assert( abs(-164.277599313 - slab.get_potential_energy()) < 1E-8 )
assert( abs(6.36379627645 - np.linalg.norm(slab.get_forces()))  < 1E-8 )

os.remove(pot_fn)