diff --git a/libs/checkpoint-sqlite/tests/test_sqlite.py b/libs/checkpoint-sqlite/tests/test_sqlite.py index f242f54f5..062bde23c 100644 --- a/libs/checkpoint-sqlite/tests/test_sqlite.py +++ b/libs/checkpoint-sqlite/tests/test_sqlite.py @@ -116,7 +116,17 @@ class TestSqliteSaver: search_results_5[1].config["configurable"]["checkpoint_ns"], } == {"", "inner"} - # TODO: test before and limit params + # search with before param + search_results_6 = list(saver.list(None, before=search_results_5[1].config)) + assert len(search_results_6) == 1 + assert search_results_6[0].config["configurable"]["thread_id"] == "thread-1" + + # search with limit param + search_results_7 = list( + saver.list({"configurable": {"thread_id": "thread-2"}}, limit=1) + ) + assert len(search_results_7) == 1 + assert search_results_7[0].config["configurable"]["thread_id"] == "thread-2" def test_search_where(self) -> None: # call method / assertions