Spaces:
Sleeping
Sleeping
| from parsers.openai_parser import OpenAIParser | |
| def run_tests(): | |
| tests = [ | |
| # --- Basic Attribution --- | |
| { | |
| "desc": "Post-quote attribution", | |
| "input": '"Hi," said Brad. "I’m home," he said.', | |
| "expected": ["[Brad]: Hi,", "[Brad]: I’m home,"] | |
| }, | |
| { | |
| "desc": "Narration-based last seen", | |
| "input": "Brad entered. \"I’m home,\" he said.", | |
| "expected": ["[Brad]: I’m home,"] | |
| }, | |
| { | |
| "desc": "Simple pronoun resolution", | |
| "input": '"Hi," said Zara. Nathaniel looked at him and said "Hey". "What\'s up?" she replied.', | |
| "expected": ["[Zara]: Hi,", "[Nathaniel]: Hey", "[Zara]: What's up?"] | |
| }, | |
| { | |
| "desc": "Fallback to narrator", | |
| "input": "\"Hello there.\"", | |
| "expected": ["[Narrator]: Hello there."] | |
| }, | |
| # --- Multiple Characters --- | |
| { | |
| "desc": "Conversation back and forth", | |
| "input": '"Hello," said Brad. "Hi," said Zara. "How are you?" he asked.', | |
| "expected": ["[Brad]: Hello,", "[Zara]: Hi,", "[Brad]: How are you?"] | |
| }, | |
| { | |
| "desc": "Ambiguous multiple males → fallback", | |
| "input": '"Hey," said Brad. Nathaniel walked in. "Yo," he said.', | |
| "expected": ["[Brad]: Hey,", "[Narrator]: Yo,"] | |
| }, | |
| # --- Punctuation Handling --- | |
| { | |
| "desc": "British punctuation", | |
| "input": '"Hello there." said Brad.', | |
| "expected": ["[Brad]: Hello there."] | |
| }, | |
| { | |
| "desc": "US punctuation", | |
| "input": '"Hello there", said Brad.', | |
| # If parser strips comma → update expected | |
| "expected": ["[Brad]: Hello there,"] | |
| }, | |
| # --- Emotion/Adverb Inference --- | |
| { | |
| "desc": "Verb to emotion mapping", | |
| "input": '"Go away!" shouted Brad.', | |
| "expected": ["[Brad](angry): Go away!"] | |
| }, | |
| { | |
| "desc": "Adverb to emotion mapping", | |
| "input": '"I’m fine," Brad said softly.', | |
| "expected": ["[Brad](gentle): I’m fine,"] | |
| }, | |
| # --- Complex Narration --- | |
| { | |
| "desc": "Inline narration updates last seen", | |
| "input": '"Hi," said Zara. Nathaniel smiled. "Hey there," he said.', | |
| "expected": ["[Zara]: Hi,", "[Nathaniel]: Hey there,"] | |
| }, | |
| { | |
| "desc": "Pronoun chain with missing female (→ fallback)", | |
| "input": 'Brad walked in. "Hello," he said. "How are you?" she asked. "I’m fine," he replied.', | |
| "expected": ["[Brad]: Hello,", "[Narrator]: How are you?", "[Brad]: I’m fine,"] | |
| }, | |
| ] | |
| for include_narration in (True, False): | |
| print("\n===============================") | |
| print(f"Testing with include_narration={include_narration}") | |
| print("===============================") | |
| parser = OpenAIParser( | |
| include_narration=include_narration, detect_fx=True) | |
| for t in tests: | |
| result = parser.convert(t["input"]) | |
| print(f"\n--- {t['desc']} ---") | |
| for line in result.formatted_text.split("\n"): | |
| print(line) | |
| print("Expected:", t["expected"]) | |
| if not include_narration: | |
| print("(Narration lines should be suppressed here)") | |
| if __name__ == "__main__": | |
| run_tests() | |