Testing retrieval of 'pre_buffer'.

yohplala · Jun 18, 2024 · ebf7920 · ebf7920
1 parent b627175
commit ebf7920
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/test_aggstream/test_aggstream_advanced.py b/tests/test_aggstream/test_aggstream_advanced.py
@@ -406,6 +406,22 @@ def check(on, buffer):
     assert streamagg_md_key1[KEY_PRE_BUFFER] == pre_buffer_ref
     streamagg_md_key2 = store[key2]._oups_metadata[KEY_AGGSTREAM]
     assert streamagg_md_key2[KEY_PRE_BUFFER] == pre_buffer_ref
+    # Testing retrieval of 'pre_buffer'.
+    as_2 = AggStream(
+        ordered_on=ordered_on,
+        store=store,
+        keys={
+            filter1: {key1: deepcopy(key1_cf)},
+            filter2: {key2: deepcopy(key2_cf)},
+        },
+        filters={
+            filter1: [(filter_on, "==", True)],
+            filter2: [(filter_on, "==", False)],
+        },
+        max_row_group_size=max_row_group_size,
+        pre=check,
+    )
+    assert as_2.seed_config[KEY_PRE_BUFFER] == pre_buffer_ref
 
 
 def post(buffer: dict, bin_res: DataFrame, snap_res: DataFrame):