Skip to content

Commit 56739ba

Browse files
Fix ring buffer benchmarks
The OrderedRingBuffer class was previously updated to use datetime aware objects. Thus, the new ring buffer benchmarks needs to be updated to fix type errors as arithmetic operation cannot be performed using both naive and aware datetimes. Signed-off-by: Daniel Zullo <[email protected]>
1 parent 914716c commit 56739ba

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)