test for time_buffer

open-aviation · Mar 20, 2024 · a411eba · a411eba
1 parent 8636cd4
commit a411eba
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_trino.py b/tests/test_trino.py
@@ -231,6 +231,16 @@ def test_complex_queries() -> None:
     assert df.callsign.iloc[0] == "THY5HT"
     assert df.icao24.iloc[0] == "4bb1c5"
 
+    df = trino.history(
+        start="2024-03-16 09:00",
+        stop="2024-03-16 11:00",
+        time_buffer='25m',
+        airport="UGTB",
+        bounds=(44.958636,41.665760,44.965417,41.670505)
+    )
+    assert df is not None
+    assert len(df.groupby(["icao24", "callsign"])) == 4
+
 
 def test_specific_columns() -> None:
     df = trino.history(