import unittest
import io
import sys


def _dynamic_test(test_case, cond, ok, fail):
    if cond:
        test_case._testMethodName = ok
        test_case.assertTrue(True)
    else:
        test_case._testMethodName = fail
        test_case.fail(fail)


def normalize(text):
    return " ".join(text.split())


class TestUserCode(unittest.TestCase):

    def test_import(self):
        with open("user_code.py") as f:
            src = f.read()

        cond = "from bs4 import BeautifulSoup" in src

        _dynamic_test(
            self,
            cond,
            "Imported BeautifulSoup.",
            "BeautifulSoup not imported."
        )

    def test_soup_exists(self):
        import user_code
        cond = hasattr(user_code, "soup")

        _dynamic_test(self, cond, "`soup` exists.", "`soup` missing.")

    def test_soup_valid(self):
        import user_code
        from bs4 import BeautifulSoup

        try:
            cond = isinstance(user_code.soup, BeautifulSoup)
        except:
            cond = False

        _dynamic_test(self, cond, "`soup` ok.", "`soup` invalid.")

    def test_html_string(self):
        import user_code

        try:
            cond = isinstance(user_code.html, str)
        except:
            cond = False

        _dynamic_test(self, cond, "`html` ok.", "`html` invalid.")

    def test_output(self):
        import user_code
        import importlib

        buf = io.StringIO()
        sys.stdout = buf
        importlib.reload(user_code)
        sys.stdout = sys.__stdout__

        lines = buf.getvalue().splitlines()

        if len(lines) < 2:
            _dynamic_test(self, False, "", "Not enough output.")
            return

        # First line: p attributes
        printed_attrs = lines[0].strip()
        expected_attrs = str(user_code.soup.p.attrs)

        if printed_attrs != expected_attrs:
            _dynamic_test(self, False, "", "Wrong p attrs.")
            return

        # Remaining lines: UL text
        printed_ul_block = " ".join([ln.strip() for ln in lines[1:] if ln.strip()])
        printed_ul_norm = normalize(printed_ul_block)

        expected_ul_norm = normalize(user_code.soup.ul.get_text())

        if printed_ul_norm != expected_ul_norm:
            _dynamic_test(self, False, "", "Wrong UL text.")
            return

        _dynamic_test(self, True, "Correct output.", "")


if __name__ == "__main__":
    unittest.main()


test_code.py

Aprenda a extrair dados valiosos de sites automaticamente. Domine técnicas para coletar informações de forma eficiente, possibilitando análises aprofundadas, decisões orientadas por dados e a descoberta de novos insights a partir do vasto oceano de informações online.

Explore a estrutura de um arquivo HTML, aprenda a carregá-lo de forma eficiente e adquira habilidades práticas para trabalhar com os dados contidos nele.

Explore os conceitos básicos do Beautiful Soup para extrair dados de documentos HTML. Aprenda a navegar pela estrutura do HTML, acessar elementos específicos e trabalhar com seus elementos filhos para construir uma base sólida em web scraping.

Domine o trabalho com atributos de elementos e conteúdo utilizando o Beautiful Soup. Adquira habilidades para extrair informações específicas, buscar elementos por valores de atributos e aplicar técnicas avançadas para analisar e processar dados HTML de forma eficiente.

Desafio: Atributos

Solução