Spaces:
Paused
Paused
| import pytest | |
| from unittest.mock import patch | |
| from bs4.builder import DetectsXMLParsedAsHTML | |
| class TestDetectsXMLParsedAsHTML(object): | |
| def test_warn_if_markup_looks_like_xml(self, markup, looks_like_xml): | |
| # Test of our ability to guess at whether markup looks XML-ish | |
| # _and_ not HTML-ish. | |
| with patch('bs4.builder.DetectsXMLParsedAsHTML._warn') as mock: | |
| for data in markup, markup.encode('utf8'): | |
| result = DetectsXMLParsedAsHTML.warn_if_markup_looks_like_xml( | |
| data | |
| ) | |
| assert result == looks_like_xml | |
| if looks_like_xml: | |
| assert mock.called | |
| else: | |
| assert not mock.called | |
| mock.reset_mock() | |