From 150ac0cb79d5019b033da433b6e35abab28b099a Mon Sep 17 00:00:00 2001 From: Christophe Bornet Date: Fri, 4 Apr 2025 19:43:47 +0200 Subject: [PATCH] core: Add ruff rules DTZ (#30657) Add ruff rules DTZ: https://docs.astral.sh/ruff/rules/#flake8-datetimez-dtz --- libs/core/pyproject.toml | 1 - .../document_loaders/test_langsmith.py | 6 +- .../indexing/test_in_memory_record_manager.py | 106 +++++++--- .../unit_tests/indexing/test_indexing.py | 198 +++++++++++++----- 4 files changed, 232 insertions(+), 79 deletions(-) diff --git a/libs/core/pyproject.toml b/libs/core/pyproject.toml index 4e206731576..0f19d7b4f59 100644 --- a/libs/core/pyproject.toml +++ b/libs/core/pyproject.toml @@ -99,7 +99,6 @@ ignore = [ "ARG", "BLE", "ERA", - "DTZ", "FBT001", "FBT002", "PGH", diff --git a/libs/core/tests/unit_tests/document_loaders/test_langsmith.py b/libs/core/tests/unit_tests/document_loaders/test_langsmith.py index 6c4c7a54170..a34d031b0ef 100644 --- a/libs/core/tests/unit_tests/document_loaders/test_langsmith.py +++ b/libs/core/tests/unit_tests/document_loaders/test_langsmith.py @@ -18,21 +18,21 @@ EXAMPLES = [ outputs={"res": "a"}, dataset_id=uuid.uuid4(), id=uuid.uuid4(), - created_at=datetime.datetime.now(), + created_at=datetime.datetime.now(datetime.timezone.utc), ), Example( inputs={"first": {"second": "bar"}}, outputs={"res": "b"}, dataset_id=uuid.uuid4(), id=uuid.uuid4(), - created_at=datetime.datetime.now(), + created_at=datetime.datetime.now(datetime.timezone.utc), ), Example( inputs={"first": {"second": "baz"}}, outputs={"res": "c"}, dataset_id=uuid.uuid4(), id=uuid.uuid4(), - created_at=datetime.datetime.now(), + created_at=datetime.datetime.now(datetime.timezone.utc), ), ] diff --git a/libs/core/tests/unit_tests/indexing/test_in_memory_record_manager.py b/libs/core/tests/unit_tests/indexing/test_in_memory_record_manager.py index 8c7e1767ccd..991c66634c6 100644 --- a/libs/core/tests/unit_tests/indexing/test_in_memory_record_manager.py +++ b/libs/core/tests/unit_tests/indexing/test_in_memory_record_manager.py @@ -1,4 +1,4 @@ -from datetime import datetime +from datetime import datetime, timezone from unittest.mock import patch import pytest @@ -55,50 +55,94 @@ def test_update_timestamp(manager: InMemoryRecordManager) -> None: """Test updating records in the database.""" # no keys should be present in the set with patch.object( - manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): manager.update(["key1"]) assert manager.list_keys() == ["key1"] - assert manager.list_keys(before=datetime(2021, 1, 1).timestamp()) == [] - assert manager.list_keys(after=datetime(2021, 1, 1).timestamp()) == ["key1"] - assert manager.list_keys(after=datetime(2021, 1, 3).timestamp()) == [] + assert ( + manager.list_keys(before=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp()) + == [] + ) + assert manager.list_keys( + after=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp() + ) == ["key1"] + assert ( + manager.list_keys(after=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp()) + == [] + ) # Update the timestamp with patch.object( - manager, "get_time", return_value=datetime(2023, 1, 5).timestamp() + manager, + "get_time", + return_value=datetime(2023, 1, 5, tzinfo=timezone.utc).timestamp(), ): manager.update(["key1"]) assert manager.list_keys() == ["key1"] - assert manager.list_keys(before=datetime(2023, 1, 1).timestamp()) == [] - assert manager.list_keys(after=datetime(2023, 1, 1).timestamp()) == ["key1"] - assert manager.list_keys(after=datetime(2023, 1, 3).timestamp()) == ["key1"] + assert ( + manager.list_keys(before=datetime(2023, 1, 1, tzinfo=timezone.utc).timestamp()) + == [] + ) + assert manager.list_keys( + after=datetime(2023, 1, 1, tzinfo=timezone.utc).timestamp() + ) == ["key1"] + assert manager.list_keys( + after=datetime(2023, 1, 3, tzinfo=timezone.utc).timestamp() + ) == ["key1"] async def test_aupdate_timestamp(manager: InMemoryRecordManager) -> None: """Test updating records in the database.""" # no keys should be present in the set with patch.object( - manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): await manager.aupdate(["key1"]) assert await manager.alist_keys() == ["key1"] - assert await manager.alist_keys(before=datetime(2021, 1, 1).timestamp()) == [] - assert await manager.alist_keys(after=datetime(2021, 1, 1).timestamp()) == ["key1"] - assert await manager.alist_keys(after=datetime(2021, 1, 3).timestamp()) == [] + assert ( + await manager.alist_keys( + before=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp() + ) + == [] + ) + assert await manager.alist_keys( + after=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp() + ) == ["key1"] + assert ( + await manager.alist_keys( + after=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp() + ) + == [] + ) # Update the timestamp with patch.object( - manager, "get_time", return_value=datetime(2023, 1, 5).timestamp() + manager, + "get_time", + return_value=datetime(2023, 1, 5, tzinfo=timezone.utc).timestamp(), ): await manager.aupdate(["key1"]) assert await manager.alist_keys() == ["key1"] - assert await manager.alist_keys(before=datetime(2023, 1, 1).timestamp()) == [] - assert await manager.alist_keys(after=datetime(2023, 1, 1).timestamp()) == ["key1"] - assert await manager.alist_keys(after=datetime(2023, 1, 3).timestamp()) == ["key1"] + assert ( + await manager.alist_keys( + before=datetime(2023, 1, 1, tzinfo=timezone.utc).timestamp() + ) + == [] + ) + assert await manager.alist_keys( + after=datetime(2023, 1, 1, tzinfo=timezone.utc).timestamp() + ) == ["key1"] + assert await manager.alist_keys( + after=datetime(2023, 1, 3, tzinfo=timezone.utc).timestamp() + ) == ["key1"] def test_exists(manager: InMemoryRecordManager) -> None: @@ -138,14 +182,18 @@ async def test_list_keys(manager: InMemoryRecordManager) -> None: assert await manager.alist_keys() == [] with patch.object( - manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): manager.update(["key1", "key2"]) manager.update(["key3"], group_ids=["group1"]) manager.update(["key4"], group_ids=["group2"]) with patch.object( - manager, "get_time", return_value=datetime(2021, 1, 10).timestamp() + manager, + "get_time", + return_value=datetime(2021, 1, 10, tzinfo=timezone.utc).timestamp(), ): manager.update(["key5"]) @@ -163,14 +211,18 @@ async def test_list_keys(manager: InMemoryRecordManager) -> None: assert await manager.alist_keys(group_ids=["group1"]) == ["key3"] # Before - assert sorted(manager.list_keys(before=datetime(2021, 1, 3).timestamp())) == [ + assert sorted( + manager.list_keys(before=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp()) + ) == [ "key1", "key2", "key3", "key4", ] assert sorted( - await manager.alist_keys(before=datetime(2021, 1, 3).timestamp()) + await manager.alist_keys( + before=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp() + ) ) == [ "key1", "key2", @@ -179,10 +231,14 @@ async def test_list_keys(manager: InMemoryRecordManager) -> None: ] # After - assert sorted(manager.list_keys(after=datetime(2021, 1, 3).timestamp())) == ["key5"] - assert sorted(await manager.alist_keys(after=datetime(2021, 1, 3).timestamp())) == [ - "key5" - ] + assert sorted( + manager.list_keys(after=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp()) + ) == ["key5"] + assert sorted( + await manager.alist_keys( + after=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp() + ) + ) == ["key5"] results = manager.list_keys(limit=1) assert len(results) == 1 diff --git a/libs/core/tests/unit_tests/indexing/test_indexing.py b/libs/core/tests/unit_tests/indexing/test_indexing.py index 6f7db250c63..683d25d2cf4 100644 --- a/libs/core/tests/unit_tests/indexing/test_indexing.py +++ b/libs/core/tests/unit_tests/indexing/test_indexing.py @@ -1,5 +1,5 @@ from collections.abc import AsyncIterator, Iterable, Iterator, Sequence -from datetime import datetime +from datetime import datetime, timezone from typing import ( Any, ) @@ -151,7 +151,9 @@ def test_index_simple_delete_full( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert index(loader, record_manager, vector_store, cleanup="full") == { "num_added": 2, @@ -161,7 +163,9 @@ def test_index_simple_delete_full( } with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert index(loader, record_manager, vector_store, cleanup="full") == { "num_added": 0, @@ -182,7 +186,9 @@ def test_index_simple_delete_full( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): indexing_result = index(loader, record_manager, vector_store, cleanup="full") @@ -202,7 +208,9 @@ def test_index_simple_delete_full( # Attempt to index again verify that nothing changes with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert index(loader, record_manager, vector_store, cleanup="full") == { "num_added": 0, @@ -228,7 +236,9 @@ async def test_aindex_simple_delete_full( ) with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert await aindex(loader, arecord_manager, vector_store, cleanup="full") == { "num_added": 2, @@ -238,7 +248,9 @@ async def test_aindex_simple_delete_full( } with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert await aindex(loader, arecord_manager, vector_store, cleanup="full") == { "num_added": 0, @@ -259,7 +271,9 @@ async def test_aindex_simple_delete_full( ) with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert await aindex(loader, arecord_manager, vector_store, cleanup="full") == { "num_added": 1, @@ -277,7 +291,9 @@ async def test_aindex_simple_delete_full( # Attempt to index again verify that nothing changes with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert await aindex(loader, arecord_manager, vector_store, cleanup="full") == { "num_added": 0, @@ -303,7 +319,9 @@ def test_index_delete_full_recovery_after_deletion_failure( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert index(loader, record_manager, vector_store, cleanup="full") == { "num_added": 2, @@ -325,7 +343,9 @@ def test_index_delete_full_recovery_after_deletion_failure( with ( patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ), patch.object(vector_store, "delete", return_value=False), pytest.raises(IndexingException), @@ -346,7 +366,9 @@ def test_index_delete_full_recovery_after_deletion_failure( } with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 3).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp(), ): indexing_result = index(loader, record_manager, vector_store, cleanup="full") doc_texts = { @@ -380,7 +402,9 @@ async def test_aindex_delete_full_recovery_after_deletion_failure( ) with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert await aindex(loader, arecord_manager, vector_store, cleanup="full") == { "num_added": 2, @@ -402,7 +426,9 @@ async def test_aindex_delete_full_recovery_after_deletion_failure( with ( patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ), patch.object(vector_store, "adelete", return_value=False), pytest.raises(IndexingException), @@ -425,7 +451,9 @@ async def test_aindex_delete_full_recovery_after_deletion_failure( } with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 3).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp(), ): indexing_result = await aindex( loader, arecord_manager, vector_store, cleanup="full" @@ -564,7 +592,9 @@ def test_index_simple_delete_scoped_full( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -580,7 +610,9 @@ def test_index_simple_delete_scoped_full( } with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -609,7 +641,9 @@ def test_index_simple_delete_scoped_full( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 3).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -636,7 +670,9 @@ def test_index_simple_delete_scoped_full( # Attempt to index again verify that nothing changes with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 4).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 4, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -678,7 +714,9 @@ async def test_aindex_simple_delete_scoped_full( ) with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader, @@ -694,7 +732,9 @@ async def test_aindex_simple_delete_scoped_full( } with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader, @@ -723,7 +763,9 @@ async def test_aindex_simple_delete_scoped_full( ) with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 3).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader, @@ -750,7 +792,9 @@ async def test_aindex_simple_delete_scoped_full( # Attempt to index again verify that nothing changes with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 4).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 4, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader, @@ -880,7 +924,9 @@ def test_index_empty_doc_scoped_full( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -896,7 +942,9 @@ def test_index_empty_doc_scoped_full( } with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -914,7 +962,9 @@ def test_index_empty_doc_scoped_full( loader = ToyLoader(documents=[]) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 3).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -956,7 +1006,9 @@ async def test_aindex_empty_doc_scoped_full( ) with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader, @@ -972,7 +1024,9 @@ async def test_aindex_empty_doc_scoped_full( } with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader, @@ -990,7 +1044,9 @@ async def test_aindex_empty_doc_scoped_full( loader = ToyLoader(documents=[]) with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 3).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader, @@ -1024,7 +1080,9 @@ def test_no_delete( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1041,7 +1099,9 @@ def test_no_delete( # If we add the same content twice it should be skipped with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1071,7 +1131,9 @@ def test_no_delete( # Should result in no updates or deletions! with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1105,7 +1167,9 @@ async def test_ano_delete( ) with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader, @@ -1122,7 +1186,9 @@ async def test_ano_delete( # If we add the same content twice it should be skipped with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader, @@ -1152,7 +1218,9 @@ async def test_ano_delete( # Should result in no updates or deletions! with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader, @@ -1186,7 +1254,9 @@ def test_incremental_delete( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1210,7 +1280,9 @@ def test_incremental_delete( # Attempt to index again verify that nothing changes with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1245,7 +1317,9 @@ def test_incremental_delete( # Attempt to index again verify that nothing changes with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 3).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1290,7 +1364,9 @@ def test_incremental_delete_with_same_source( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1323,7 +1399,9 @@ def test_incremental_delete_with_same_source( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1374,7 +1452,9 @@ def test_incremental_indexing_with_batch_size( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1398,7 +1478,9 @@ def test_incremental_indexing_with_batch_size( assert doc_texts == {"1", "2", "3", "4"} with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1448,7 +1530,9 @@ def test_incremental_delete_with_batch_size( ) with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 1).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 1, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1473,7 +1557,9 @@ def test_incremental_delete_with_batch_size( # Attempt to index again verify that nothing changes with patch.object( - record_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + record_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert index( loader, @@ -1498,7 +1584,9 @@ def test_incremental_delete_with_batch_size( # Attempt to index again verify that nothing changes with patch.object( - record_manager, "get_time", return_value=datetime(2022, 1, 3).timestamp() + record_manager, + "get_time", + return_value=datetime(2022, 1, 3, tzinfo=timezone.utc).timestamp(), ): # Docs with same content docs = [ @@ -1534,7 +1622,9 @@ def test_incremental_delete_with_batch_size( # Attempt to index again verify that nothing changes with patch.object( - record_manager, "get_time", return_value=datetime(2023, 1, 4).timestamp() + record_manager, + "get_time", + return_value=datetime(2023, 1, 4, tzinfo=timezone.utc).timestamp(), ): # Docs with same content docs = [ @@ -1570,7 +1660,9 @@ def test_incremental_delete_with_batch_size( # Try to index with changed docs now with patch.object( - record_manager, "get_time", return_value=datetime(2024, 1, 5).timestamp() + record_manager, + "get_time", + return_value=datetime(2024, 1, 5, tzinfo=timezone.utc).timestamp(), ): # Docs with same content docs = [ @@ -1622,7 +1714,9 @@ async def test_aincremental_delete( ) with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader.lazy_load(), @@ -1646,7 +1740,9 @@ async def test_aincremental_delete( # Attempt to index again verify that nothing changes with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 2).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 2, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader.lazy_load(), @@ -1681,7 +1777,9 @@ async def test_aincremental_delete( # Attempt to index again verify that nothing changes with patch.object( - arecord_manager, "get_time", return_value=datetime(2021, 1, 3).timestamp() + arecord_manager, + "get_time", + return_value=datetime(2021, 1, 3, tzinfo=timezone.utc).timestamp(), ): assert await aindex( loader.lazy_load(),