Skip to content

Commit 4a80dac

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 0601473 commit 4a80dac

5 files changed

Lines changed: 28 additions & 17 deletions

File tree

src/cachier/cores/mongo.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -222,13 +222,15 @@ def _evict_lru_entries(self, current_size: int) -> None:
222222
# For entries without last_access, use the time field as fallback
223223
pipeline = [
224224
{"$match": {"func": self._func_str, "size": {"$exists": True}}},
225-
{"$addFields": {
226-
"sort_time": {
227-
"$ifNull": ["$last_access", "$time"]
225+
{
226+
"$addFields": {
227+
"sort_time": {"$ifNull": ["$last_access", "$time"]}
228228
}
229-
}},
230-
{"$sort": {"sort_time": 1}}, # Sort by sort_time ascending (oldest first)
231-
{"$project": {"key": 1, "size": 1}}
229+
},
230+
{
231+
"$sort": {"sort_time": 1}
232+
}, # Sort by sort_time ascending (oldest first)
233+
{"$project": {"key": 1, "size": 1}},
232234
]
233235
entries = self.mongo_collection.aggregate(pipeline)
234236

tests/test_cache_size_limit.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,10 +128,10 @@ def test_cache_size_within_limit_mongo():
128128
try:
129129
mongo_db = mongo_client["cachier_test"]
130130
mongo_collection = mongo_db["test_cache_size_within_limit"]
131-
131+
132132
# Clear collection before test
133133
mongo_collection.delete_many({})
134-
134+
135135
call_count = 0
136136

137137
@cachier.cachier(
@@ -148,7 +148,7 @@ def func(x):
148148
func(2)
149149
func(3)
150150
assert call_count == 3
151-
151+
152152
# All should be cached
153153
func(1)
154154
func(2)

tests/test_defaults.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,11 @@
77
from dataclasses import replace
88

99
import pytest
10+
1011
import cachier
1112
from tests.test_mongo_core import (
12-
_test_mongetter,
1313
_get_mongetter_by_collection_name,
14+
_test_mongetter,
1415
)
1516

1617
MONGO_DELTA = datetime.timedelta(seconds=3)

tests/test_entry_size_limit.py

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -45,13 +45,15 @@ def test_entry_size_limit_not_cached_mongo():
4545
try:
4646
mongo_db = mongo_client["cachier_test"]
4747
mongo_collection = mongo_db["test_entry_size_not_cached"]
48-
48+
4949
# Clear collection before test
5050
mongo_collection.delete_many({})
51-
51+
5252
call_count = 0
5353

54-
@cachier.cachier(mongetter=lambda: mongo_collection, entry_size_limit="10B")
54+
@cachier.cachier(
55+
mongetter=lambda: mongo_collection, entry_size_limit="10B"
56+
)
5557
def func(x):
5658
nonlocal call_count
5759
call_count += 1
@@ -61,7 +63,9 @@ def func(x):
6163
val1 = func(1)
6264
val2 = func(1)
6365
assert val1 == val2
64-
assert call_count == 2 # Should be called twice since value is too large to cache
66+
assert (
67+
call_count == 2
68+
) # Should be called twice since value is too large to cache
6569
finally:
6670
mongo_client.close()
6771

@@ -74,13 +78,15 @@ def test_entry_size_limit_cached_mongo():
7478
try:
7579
mongo_db = mongo_client["cachier_test"]
7680
mongo_collection = mongo_db["test_entry_size_cached"]
77-
81+
7882
# Clear collection before test
7983
mongo_collection.delete_many({})
80-
84+
8185
call_count = 0
8286

83-
@cachier.cachier(mongetter=lambda: mongo_collection, entry_size_limit="1KB")
87+
@cachier.cachier(
88+
mongetter=lambda: mongo_collection, entry_size_limit="1KB"
89+
)
8490
def func(x):
8591
nonlocal call_count
8692
call_count += 1

tests/test_mongo_core.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,7 @@ def _get_mongetter_by_collection_name(collection_name=_COLLECTION_NAME):
126126
127127
This is important for preventing cache conflicts when running tests in
128128
parallel.
129+
129130
"""
130131

131132
def _custome_mongetter():
@@ -691,6 +692,7 @@ def test_func(x):
691692
def test_mongo_unsupported_replacement_policy():
692693
"""Test that unsupported replacement policy raises ValueError."""
693694
import pymongo
695+
694696
from cachier.cores.mongo import _MongoCore
695697

696698
# Clear before test

0 commit comments

Comments
 (0)