langchain/libs/community/tests/unit_tests/chat_loaders/test_whatsapp.py

import pathlib

from langchain_community.chat_loaders import utils, whatsapp


def test_whatsapp_chat_loader() -> None:
    chat_path = pathlib.Path(__file__).parent / "data" / "whatsapp_chat.txt"
    loader = whatsapp.WhatsAppChatLoader(str(chat_path))

    chat_sessions = list(
        utils.map_ai_messages(loader.lazy_load(), sender="Dr. Feather")
    )
    assert chat_sessions, "Chat sessions should not be empty"

    assert chat_sessions[0]["messages"], "Chat messages should not be empty"

    assert (
        "I spotted a rare Hyacinth Macaw yesterday in the Amazon Rainforest."
        " Such a magnificent creature!" in chat_sessions[0]["messages"][0].content
    ), "Chat content mismatch"