import unittest
import io
import sys


def _dynamic(test, cond, ok, fail):
    if cond:
        test._testMethodName = ok
        test.assertTrue(True)
    else:
        test._testMethodName = fail
        test.fail(fail)


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


class TestUserCode(unittest.TestCase):

    def test_import(self):
        with open("user_code.py") as f:
            src = f.read()
        _dynamic(
            self,
            "from bs4 import BeautifulSoup" in src,
            "Imported BeautifulSoup.",
            "BeautifulSoup not imported."
        )

    def test_soup_exists(self):
        import user_code
        _dynamic(self, hasattr(user_code, "soup"), "`soup` exists.", "`soup` missing.")

    def test_soup_valid(self):
        import user_code
        from bs4 import BeautifulSoup
        try:
            ok = isinstance(user_code.soup, BeautifulSoup)
        except:
            ok = False
        _dynamic(self, ok, "`soup` ok.", "`soup` invalid.")

    def test_html_ok(self):
        import user_code
        try:
            ok = isinstance(user_code.html, str)
        except:
            ok = False
        _dynamic(self, ok, "`html` ok.", "`html` invalid.")

    def test_output(self):
        import user_code
        import importlib

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

        output = buf.getvalue()

        if not output.strip():
            _dynamic(self, False, "", "No output.")
            return

        # Expected values
        first_div = str(user_code.soup.find("div"))
        all_p = str(user_code.soup.find_all("p", {"id": "id0"}))

        out_norm = norm(output)
        div_norm = norm(first_div)
        p_norm = norm(all_p)

        # Check div exists
        if div_norm not in out_norm:
            _dynamic(self, False, "", "First div not printed.")
            return

        # Check p[id=id0] exists
        if p_norm not in out_norm:
            _dynamic(self, False, "", "id0 p-tags not printed.")
            return

        # Ensure order
        if out_norm.find(div_norm) > out_norm.find(p_norm):
            _dynamic(self, False, "", "Wrong print order.")
            return

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


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


test_code.py

Learn to extract valuable data from websites automatically. Master techniques to gather information efficiently, enabling you to perform in-depth analysis, make data-driven decisions, and unlock new insights from the vast ocean of online information.

Explore the structure of an HTML file, learn how to load it efficiently, and gain practical skills in working with the data it contains.

Explore the basics of Beautiful Soup to extract data from HTML documents. Learn how to navigate the structure of HTML, access specific elements, and work with their child elements to build a strong foundation in web scraping.

Master working with element attributes and content using Beautiful Soup. Gain the skills to extract specific information, search elements by attribute values, and use advanced techniques to efficiently parse and analyze HTML data.

Challenge: Find All

解答