"""Tests for source distinction - SOUL.md compliance."""
import pytest


class TestSourceDistinction:
    def test_verified_claim(self):
        from scripts.source_distinction import verified, SourceType
        claim = verified("Paris is the capital", "web_search:Paris")
        assert claim.source_type == SourceType.VERIFIED
        assert claim.source_ref == "web_search:Paris"
        assert claim.confidence == 0.95

    def test_inferred_claim(self):
        from scripts.source_distinction import inferred, SourceType
        claim = inferred("this approach is better")
        assert claim.source_type == SourceType.INFERRED
        assert claim.hedging == "I think"

    def test_stated_claim(self):
        from scripts.source_distinction import stated, SourceType
        claim = stated("my name is Alexander")
        assert claim.source_type == SourceType.STATED
        assert claim.confidence == 1.0

    def test_render_verified(self):
        from scripts.source_distinction import annotate_response, verified
        resp = annotate_response("test", [verified("Paris is capital", "web")])
        rendered = resp.render()
        assert "[verified: web]" in rendered

    def test_render_inferred(self):
        from scripts.source_distinction import annotate_response, inferred
        resp = annotate_response("test", [ inferred("this is better")])
        rendered = resp.render()
        assert "I think" in rendered

    def test_counts(self):
        from scripts.source_distinction import annotate_response, verified, inferred
        resp = annotate_response("test", [
            verified("a", "src"), verified("b", "src"), inferred("c"),
        ])
        assert resp.verified_count == 2
        assert resp.inferred_count == 1

    def test_hedging_detection(self):
        from scripts.source_distinction import source_distinction_check
        result = source_distinction_check("I think this is probably right, but I believe it could be different")
        assert result["has_hedging"]
        assert result["hedging_count"] >= 3

    def test_no_hedging(self):
        from scripts.source_distinction import source_distinction_check
        result = source_distinction_check("The capital of France is Paris.")
        assert not result["has_hedging"]

    def test_format_for_display(self):
        from scripts.source_distinction import format_for_display, annotate_response, verified, inferred
        resp = annotate_response("test", [verified("a", "src"), inferred("b")])
        output = format_for_display(resp)
        assert "=" in output  # verified icon
        assert "~" in output  # inferred icon