Skip to content

Commit b484d7b

Browse files
committed
Update test for sampling frequency.
1 parent af5e84f commit b484d7b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/btrdb_integration/test_streamset.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def test_streamset_arrow_windows_vs_windows_agg_all(conn, tmp_collection):
152152
assert (
153153
len(
154154
other_arrow_df.filter(
155-
regex=".*\/[min,mean,max,count,stddev]", axis=1
155+
regex=r".*\/[min,mean,max,count,stddev]", axis=1
156156
).columns
157157
)
158158
== 3 * 5
@@ -487,6 +487,7 @@ def test_timesnap_with_different_sampling_frequencies(freq, conn, tmp_collection
487487
df = stset.filter(
488488
start=start, end=stop, sampling_frequency=freq
489489
).arrow_to_dataframe()
490+
df.set_index("time", inplace=True)
490491
total_points = df.shape[0] * df.shape[1]
491492
total_raw_pts = len(v1) * len(stset)
492493
expected_frac_of_pts = 1 if freq is None else freq / data_insert_freq

0 commit comments

Comments
 (0)