def test_trajClippingPhotonFlooding(self):
options = FakeOptionsRK()
options.pbfile = "mixed.prb"
options.configfilename = "standardRKclip.conf"
runTrajPrbAdd(options)
outdata = numpy.loadtxt(options.outtrajfile, comments = "#")
self.assertEqual(len(outdata[:, 3]), 3760)
def test_trajPhotonFlooding(self):
options = FakeOptionsRK()
options.pbfile = "mixed.prb"
options.configfilename = "standardRKmulti.conf"
runTrajPrbAdd(options)
outdata = numpy.loadtxt(options.outtrajfile, comments = "#")
self.assertAlmostEqual(outdata[:, 3].sum(), 16200. / 1000, delta = 100.0 / 1000)
def test_trajClassAssignOnly(self):
options = FakeOptionsRK()
options.pbfile = "mixed.prb"
runTrajPrbAdd(options)
outdata = numpy.loadtxt(options.outtrajfile, comments = "#")
self.assertAlmostEqual(outdata[0:1000, 3].sum(), 0.5, delta = 0.1)
self.assertAlmostEqual(outdata[1000:2000, 3].sum(), 0.25, delta = 0.1)
self.assertAlmostEqual(outdata[2000:3000, 3].sum(), 0.25, delta = 0.1)
self.assertAlmostEqual(outdata[3000:4000, 3].sum(), 0.0, delta = 0.1)
