mirror of
https://github.com/run-llama/llama-hub.git
synced 2026-07-01 20:44:00 -04:00
Emanuel Ferreira
14 lines
464 B
Python
14 lines
464 B
Python
"""Check that the string iterable loader is working as expected."""
|
|
import sys
|
|
from pathlib import Path
|
|
from llama_hub.string_iterable.base import StringIterableReader
|
|
|
|
sys.path.append(Path(__file__).parent.parent)
|
|
|
|
|
|
def test_string_iterable() -> None:
|
|
"""Check that StringIterableReader works correctly."""
|
|
reader = StringIterableReader()
|
|
documents = reader.load_data(texts=["I went to the store", "I bought an apple"])
|
|
assert len(documents) == 2
|