def test_verysimple_single_packet_loop( verysimple_numba_model, verysimple_numba_plasma, verysimple_estimators, verysimple_vpacket_collection, verysimple_packet_collection, ): packet_collection = verysimple_packet_collection vpacket_collection = verysimple_vpacket_collection numba_model = verysimple_numba_model numba_plasma = verysimple_numba_plasma numba_estimators = verysimple_estimators i = 0 r_packet = RPacket( verysimple_numba_model.r_inner[0], packet_collection.packets_input_mu[i], packet_collection.packets_input_nu[i], packet_collection.packets_input_energy[i], i, ) single_packet_loop( r_packet, numba_model, numba_plasma, numba_estimators, vpacket_collection, ) npt.assert_almost_equal(r_packet.nu, 1053057938883272.8) npt.assert_almost_equal(r_packet.mu, 0.9611146425440562) npt.assert_almost_equal(r_packet.energy, 0.10327717505563379)
def static_packet(): return RPacket( r=7.5e14, nu=0.4, mu=0.3, energy=0.9, seed=1963, index=0, )
def packet(verysimple_packet_collection): return RPacket( r=7.5e14, nu=verysimple_packet_collection.packets_input_nu[0], mu=verysimple_packet_collection.packets_input_mu[0], energy=verysimple_packet_collection.packets_input_energy[0], seed=1963, index=0, )