gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add metadata_id field to documents in integration tests

Browse source
This commit is contained in:
Leon Luithlen 2024-11-26 16:47:15 +01:00
parent 899275c25e
commit 20d721f5ca
4 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/tests/integration/documents/AudioDocument_test.py
View file

@ -27,7 +27,7 @@ TEST_TEXT = """
def test_AudioDocument(): def test_AudioDocument():
document = AudioDocument( document = AudioDocument(
id=uuid.uuid4(), name="audio-dummy-test", raw_data_location="" id=uuid.uuid4(), name="audio-dummy-test", raw_data_location="", metadata_id=uuid.uuid4()
) )
with patch.object(AudioDocument, "create_transcript", return_value=TEST_TEXT): with patch.object(AudioDocument, "create_transcript", return_value=TEST_TEXT):
for ground_truth, paragraph_data in zip( for ground_truth, paragraph_data in zip(

2
cognee/tests/integration/documents/ImageDocument_test.py
View file

@ -16,7 +16,7 @@ The commotion has attracted an audience: a murder of crows has gathered in the l
def test_ImageDocument(): def test_ImageDocument():
document = ImageDocument( document = ImageDocument(
id=uuid.uuid4(), name="image-dummy-test", raw_data_location="" id=uuid.uuid4(), name="image-dummy-test", raw_data_location="", metadata_id=uuid.uuid4()
) )
with patch.object(ImageDocument, "transcribe_image", return_value=TEST_TEXT): with patch.object(ImageDocument, "transcribe_image", return_value=TEST_TEXT):

2
cognee/tests/integration/documents/PdfDocument_test.py
View file

@ -17,7 +17,7 @@ def test_PdfDocument():
"artificial-intelligence.pdf", "artificial-intelligence.pdf",
) )
document = PdfDocument( document = PdfDocument(
id=uuid.uuid4(), name="Test document.pdf", raw_data_location=test_file_path id=uuid.uuid4(), name="Test document.pdf", raw_data_location=test_file_path, metadata_id=uuid.uuid4()
) )
for ground_truth, paragraph_data in zip( for ground_truth, paragraph_data in zip(

2
cognee/tests/integration/documents/TextDocument_test.py
View file

@ -29,7 +29,7 @@ def test_TextDocument(input_file, chunk_size):
input_file, input_file,
) )
document = TextDocument( document = TextDocument(
id=uuid.uuid4(), name=input_file, raw_data_location=test_file_path id=uuid.uuid4(), name=input_file, raw_data_location=test_file_path, metadata_id=uuid.uuid4()
) )
for ground_truth, paragraph_data in zip( for ground_truth, paragraph_data in zip(