mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
27 lines
730 B
Python
27 lines
730 B
Python
|
"""Unit tests for the Logical Fallacy chain, same format as CAI"""
|
||
|
|
from langchain_experimental.fallacy_removal.base import FallacyChain
|
||
|
|
|
||
|
|
TEXT_ONE = """ This text is bad.\
|
||
|
|
|
||
|
|
Fallacy Revision request: Make it great.\
|
||
|
|
|
||
|
|
Fallacy Revision:"""
|
||
|
|
|
||
|
|
TEXT_TWO = """ This text is bad.\n\n"""
|
||
|
|
|
||
|
|
TEXT_THREE = """ This text is bad.\
|
||
|
|
|
||
|
|
Fallacy Revision request: Make it great again.\
|
||
|
|
|
||
|
|
Fallacy Revision: Better text"""
|
||
|
|
|
||
|
|
|
||
|
|
def test_fallacy_critique_parsing() -> None:
|
||
|
|
"""Test parsing of critique text."""
|
||
|
|
for text in [TEXT_ONE, TEXT_TWO, TEXT_THREE]:
|
||
|
|
fallacy_critique = FallacyChain._parse_critique(text)
|
||
|
|
|
||
|
|
assert (
|
||
|
|
fallacy_critique.strip() == "This text is bad."
|
||
|
|
), f"Failed on {text} with {fallacy_critique}"
|