embedchain

1
import hashlib
2

3
import pytest
4

5
from embedchain.loaders.local_text import LocalTextLoader
6

7

8
@pytest.fixture
9
def text_loader():
10
    return LocalTextLoader()
11

12

13
def test_load_data(text_loader):
14
    mock_content = "This is a sample text content."
15

16
    result = text_loader.load_data(mock_content)
17

18
    assert "doc_id" in result
19
    assert "data" in result
20

21
    url = "local"
22
    assert result["data"][0]["content"] == mock_content
23

24
    assert result["data"][0]["meta_data"]["url"] == url
25

26
    expected_doc_id = hashlib.sha256((mock_content + url).encode()).hexdigest()
27
    assert result["doc_id"] == expected_doc_id
28