benchmark single packet loop

Signed-off-by: Asish Kumar <[email protected]>

benchmarks/benchmark_base.py

@@ -24,6 +24,7 @@
 from tardis.transport.montecarlo.packet_collections import (
     VPacketCollection,
 )
+from tardis.transport.montecarlo.packet_trackers import RPacketTracker
 
 
 class BenchmarkBase:
@@ -361,8 +362,8 @@ def montecarlo_configuration(self):
         return montecarlo_configuration.MonteCarloConfiguration()
 
     @property
-    def rpacket_tracker(self):
-        return self.simulation_rpacket_tracking_enabled.transport.transport_state.rpacket_tracker
+    def rpacket_tracker(self): 
+        return RPacketTracker(0)
 
     @property
     def transport_state(self):

benchmarks/transport_montecarlo_interaction.py

@@ -2,16 +2,14 @@
 Basic TARDIS Benchmark.
 """
 
-from asv_runner.benchmarks.mark import parameterize, skip_benchmark
-
 import tardis.transport.montecarlo.interaction as interaction
 from benchmarks.benchmark_base import BenchmarkBase
 from tardis.transport.montecarlo.numba_interface import (
     LineInteractionType,
 )
 
 
-@skip_benchmark
+
 class BenchmarkMontecarloMontecarloNumbaInteraction(BenchmarkBase):
     """
     Class to benchmark the numba interaction function.

benchmarks/transport_montecarlo_packet.py

@@ -143,7 +143,6 @@ def time_move_packet_across_shell_boundary_emitted(self, parameters):
             packet, delta_shell, no_of_shells
         )
 
-    @skip_benchmark
     @parameterize(
         {
             "Parameters": [

benchmarks/transport_montecarlo_single_packet_loop.py

@@ -4,23 +4,21 @@
 
 from benchmarks.benchmark_base import BenchmarkBase
 from tardis.transport.montecarlo import single_packet_loop
-from asv_runner.benchmarks.mark import skip_benchmark
 
 
 class BenchmarkMontecarloMontecarloNumbaVpacket(BenchmarkBase):
     """
     Class to benchmark the single packet loop function.
     """
 
-    @skip_benchmark
     def time_single_packet_loop(self):
         single_packet_loop.single_packet_loop(
             self.packet,
             self.verysimple_numba_radial_1d_geometry,
             self.verysimple_time_explosion,
             self.verysimple_opacity_state,
             self.estimators,
-            self.verysimple_packet_collection,
+            self.verysimple_3vpacket_collection,
             self.rpacket_tracker,
             self.montecarlo_configuration
         )