dan/memory
1
0
Fork 0
mirror of https://github.com/mruwnik/memory.git synced 2025-07-29 14:16:09 +02:00
Code Issues 1 Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
Daniel O'Connell 2025-07-24 20:24:22 +02:00
parent cf456c04d6
commit beb94375da
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/memory/common/db/models/test_source_item.py
View File

@ -3,7 +3,7 @@ from unittest.mock import patch
from typing import cast from typing import cast
import pytest import pytest
from PIL import Image from PIL import Image
from memory.common import settings, chunker, extract from memory.common import settings, chunker, extract, tokens
from memory.common.db.models.source_item import ( from memory.common.db.models.source_item import (
Chunk, Chunk,
) )
@ -610,7 +610,7 @@ def test_chunk_mixed_long_content(tmp_path):
with ( with (
patch.object(settings, "FILE_STORAGE_DIR", tmp_path), patch.object(settings, "FILE_STORAGE_DIR", tmp_path),
patch.object(chunker, "DEFAULT_CHUNK_TOKENS", 10), patch.object(chunker, "DEFAULT_CHUNK_TOKENS", 10),
patch.object(chunker, "approx_token_count", return_value=100), patch.object(tokens, "approx_token_count", return_value=100),
): # Force it to be > 2 * 10 ): # Force it to be > 2 * 10
result = chunk_mixed(long_content, []) result = chunk_mixed(long_content, [])