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)