Skip to content

Commit

Permalink
Add benchmark for the utility functions
Browse files Browse the repository at this point in the history
  • Loading branch information
Sumit112192 committed Jul 28, 2024
1 parent 2ae3a98 commit 55b291e
Showing 1 changed file with 19 additions and 3 deletions.
22 changes: 19 additions & 3 deletions benchmarks/transport_montecarlo_packet_trackers.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
from benchmarks.benchmark_base import BenchmarkBase
from tardis.transport.montecarlo.packet_trackers import (
rpacket_trackers_to_dataframe,
generate_rpacket_tracker_list,
generate_rpacket_last_interaction_tracker_list,
)


Expand All @@ -15,6 +17,20 @@ class BenchmarkTransportMontecarloPacketTrackers(BenchmarkBase):
def time_rpacket_trackers_to_dataframe(self):
sim = self.simulation_rpacket_tracking_enabled
transport_state = sim.transport.transport_state
rpacket_trackers_to_dataframe(
transport_state.rpacket_tracker
)
rpacket_trackers_to_dataframe(transport_state.rpacket_tracker)

def time_generate_rpacket_tracker_list(self, no_of_packets, length):
generate_rpacket_tracker_list(no_of_packets, length)

def time_generate_rpacket_last_interaction_tracker_list(
self, no_of_packets
):
generate_rpacket_last_interaction_tracker_list(no_of_packets)

time_generate_rpacket_tracker_list.params = ([1, 10, 50], [1, 10, 50])
time_generate_rpacket_tracker_list.param_names = ["no_of_packets", "length"]

time_generate_rpacket_last_interaction_tracker_list.params = [10, 100, 1000]
time_generate_rpacket_last_interaction_tracker_list.param_names = [
"no_of_packets"
]

0 comments on commit 55b291e

Please sign in to comment.