mirror of
https://github.com/hwchase17/langchain
synced 2024-11-08 07:10:35 +00:00
69de33e024
# Add Mastodon toots loader. Loader works either with public toots, or Mastodon app credentials. Toot text and user info is loaded. I've also added integration test for this new loader as it works with public data, and a notebook with example output run now. --------- Co-authored-by: Dev 2049 <dev.dev2049@gmail.com>
15 lines
460 B
Python
15 lines
460 B
Python
"""Tests for the Mastodon toots loader"""
|
|
from langchain.document_loaders import MastodonTootsLoader
|
|
|
|
|
|
def test_mastodon_toots_loader() -> None:
|
|
"""Test Mastodon toots loader with an external query."""
|
|
# Query the Mastodon CEO's account
|
|
loader = MastodonTootsLoader(
|
|
mastodon_accounts=["@Gargron@mastodon.social"], number_toots=1
|
|
)
|
|
docs = loader.load()
|
|
|
|
assert len(docs) == 1
|
|
assert docs[0].metadata["user_info"]["id"] == 1
|