def test_convertFromFMData(self):
with open(TESTA_E_DAT, "r") as efileA:
edataA = fmdata.readFromDatFile(efileA, pscale=self.PSCALE, fscale=self.FSCALE)
ldataA = lrfdata.convertFromFMData(edataA)
with open(TESTA_LRF, "r") as lfileB:
ldataB = lrfdata.readFromImpactFile(lfileB)
n = ldataB.n
for z in xrange(n):
self.assertTrue(numpy.allclose(ldataA.p[z], ldataB.p[z]),
msg="Incorrect position at %d: %g: expecting %g" % (z, ldataA.p[z], ldataB.p[z]))
self.assertTrue(numpy.allclose(ldataA.f[z], ldataB.f[z]),
msg="Incorrect electric field at %d: %g: expecting %g" % (z, ldataA.f[z], ldataB.f[z]))
self.assertTrue(numpy.allclose(ldataA.df[z], ldataB.df[z]),
msg="Incorrect electric field derivative at %d: %g: expecting %g" % (z, ldataA.df[z], ldataA.df[z]))
def setUp(self):
efile = open(TESTA_E_DAT, "r")
edata = fmdata.readFromDatFile(efile, pscale=0.001, fscale=2.2)
efile.close()
self.lrfdata = lrfdata.convertFromFMData(edata)
