Add tests for timesnap boundary conditions.

Author: andrewchambers

tests/btrdb_integration/test_streamset.py

@@ -115,3 +115,36 @@ def test_arrow_streamset_to_dataframe(conn, tmp_collection):
     }
     expected_df = pd.DataFrame(expected_dat, index=pd.DatetimeIndex(expected_times))
     assert values.equals(expected_df)
+
+def test_timesnap_backward_keeps_in_range(conn, tmp_collection):
+    sec = 10**9
+    tv = [
+        [int(0.75 * sec), 0.75],
+        [1 * sec, 1.0],
+        [2 * sec, 2.0],
+    ]
+    s = conn.create(new_uuid(), tmp_collection, tags={"name": "s"})
+    s.insert(tv)
+    ss = btrdb.stream.StreamSet([s]).filter(
+        start=1 * sec, end=3 * sec, sampling_frequency=1
+    )
+    values = ss.arrow_values()
+    assert [1 * sec, 2 * sec] == [t.value for t in values["time"]]
+    assert [1.0, 2.0] == [v.as_py() for v in values[tmp_collection + "/s"]]
+
+
+def test_timesnap_forward_keeps_in_range(conn, tmp_collection):
+    sec = 10**9
+    tv = [
+        [1 * sec, 1.0],
+        [2 * sec, 2.0],
+        [int(2.75 * sec), 2.75],
+    ]
+    s = conn.create(new_uuid(), tmp_collection, tags={"name": "s"})
+    s.insert(tv)
+    ss = btrdb.stream.StreamSet([s]).filter(start=1 * sec, sampling_frequency=1)
+    values = ss.filter(end=int(3.0 * sec)).arrow_values()
+    assert [1 * sec, 2 * sec] == [t.value for t in values["time"]]
+    assert [1.0, 2.0] == [v.as_py() for v in values[tmp_collection + "/s"]]
+    # Same result if skipping past end instead of to end.
+    assert values == ss.filter(end=int(2.9 * sec)).arrow_values()