KakaoQuotation/tests/test_chat_parser.py

17 lines
589 B
Python

import pytest
from resources.chat_parser import parse_chat_log
def test_parse_chat_log():
test_input = """[Test Room]
--------------- 2024년 5월 9일 목요일 ---------------
[User1] [09:00] 안녕하세요
[User2] [09:01] 반갑습니다"""
expected_output = [
{'date': '2024-05-09', 'user': 'User1', 'time': '09:00', 'message': '안녕하세요'},
{'date': '2024-05-09', 'user': 'User2', 'time': '09:01', 'message': '반갑습니다'}
]
assert parse_chat_log(test_input) == expected_output
def test_empty_input():
assert parse_chat_log("") == []