Skip to content

Commit f2eed52

Browse files
Fix ring buffer benchmarks (#324)
2 parents 914716c + 56739ba commit f2eed52

File tree

2 files changed

+14
-4
lines changed

2 files changed

+14
-4
lines changed

benchmarks/timeseries/ringbuffer_memusage.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
import argparse
99
import tracemalloc
10-
from datetime import datetime, timedelta
10+
from datetime import datetime, timedelta, timezone
1111

1212
import numpy as np
1313

@@ -71,7 +71,12 @@ def main(ringbuffer_len: int, iterations: int, gap_size: int) -> None:
7171

7272
for i in range(0, ringbuffer_len * iterations, gap_size + 1):
7373
ringbuffer.update(
74-
Sample(datetime.fromtimestamp(200 + i * FIVE_MINUTES.total_seconds()), i)
74+
Sample(
75+
datetime.fromtimestamp(
76+
200 + i * FIVE_MINUTES.total_seconds(), tz=timezone.utc
77+
),
78+
i,
79+
)
7580
)
7681

7782
# Snapshot memory allocations after ringbuffer update

benchmarks/timeseries/serializable_ringbuffer.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import fnmatch
99
import os
1010
import time
11-
from datetime import datetime, timedelta
11+
from datetime import datetime, timedelta, timezone
1212
from typing import Any
1313

1414
import numpy as np
@@ -68,7 +68,12 @@ def main() -> None:
6868

6969
for i in range(0, SIZE, 10000):
7070
ringbuffer.update(
71-
Sample(datetime.fromtimestamp(200 + i * FIVE_MINUTES.total_seconds()), i)
71+
Sample(
72+
datetime.fromtimestamp(
73+
200 + i * FIVE_MINUTES.total_seconds(), tz=timezone.utc
74+
),
75+
i,
76+
)
7277
)
7378

7479
print(

0 commit comments

Comments
 (0)