Class TestAddNoThinkRewriter

tests/operators/test_rewriters.py:144–178

Tests for AddNoThinkRewriter.

class TestAddNoThinkRewriter:
    """Tests for AddNoThinkRewriter."""

    @pytest.fixture
    def rewriter(self):
        """Create rewriter instance."""
        return AddNoThinkRewriter()

    def test_rewrite_adds_empty_think(self, rewriter):
        """Test rewrite adds empty think tag."""
        item = DataItem(
            {
                "conversations": [
                    {"from": "human", "value": "Q"},
                    {"from": "gpt", "value": "Answer without think"},
                ]
            }
        )
        result = rewriter.rewrite(item, 0)
        if result:
            assert "<think>" in result

    def test_rewrite_already_has_think(self, rewriter):
        """Test rewrite when think tag already present."""
        item = DataItem(
            {
                "conversations": [
                    {"from": "human", "value": "Q"},
                    {"from": "gpt", "value": "<think>Thinking</think>Answer"},
                ]
            }
        )
        result = rewriter.rewrite(item, 0)
        # Should not add another think tag; return None
        assert result is None


class TestNormImageTagRewriter: