import unittest
import importlib
import io
import re
import inspect
from contextlib import redirect_stdout
import user_code


def _dynamic_test(test_case, condition, success_message, failure_message):
    if condition:
        test_case._testMethodName = success_message
        test_case.assertTrue(True, success_message)
    else:
        test_case._testMethodName = failure_message
        test_case.fail(failure_message)


def strip_comments_from_source(module):
    try:
        source = inspect.getsource(module)
        lines = source.split('\\n')
        cleaned = []
        for line in lines:
            stripped = line.rstrip()
            if stripped.startswith('#'):
                continue
            if '#' in line:
                line = line.split('#')[0]
            if line.strip():
                cleaned.append(line)
        return '\\n'.join(cleaned)
    except Exception:
        return ""


class TestSimpleRestocking(unittest.TestCase):

    def normalize_text(self, text):
        text = text.lower()
        text = re.sub(r'\s+', ' ', text)
        return text.strip()

    # --- STATE TESTS ---

    def test_bread_stock(self):
        importlib.reload(user_code)
        item = user_code.inventory["Bread"]
        _dynamic_test(
            self,
            item[0] == 50,
            "Bread stock is updated to 50",
            f"Expected Bread stock 50, got {item[0]}"
        )

    def test_eggs_stock(self):
        importlib.reload(user_code)
        item = user_code.inventory["Eggs"]
        _dynamic_test(
            self,
            item[0] == 200,
            "Eggs stock is updated to 200",
            f"Expected Eggs stock 200, got {item[0]}"
        )

    def test_milk_stock(self):
        importlib.reload(user_code)
        item = user_code.inventory["Milk"]
        _dynamic_test(
            self,
            item[0] == 100,
            "Milk stock is updated to 100",
            f"Expected Milk stock 100, got {item[0]}"
        )

    def test_apples_stock(self):
        importlib.reload(user_code)
        item = user_code.inventory["Apples"]
        _dynamic_test(
            self,
            item[0] == 60,
            "Apples stock is updated to 60",
            f"Expected Apples stock 60, got {item[0]}"
        )

    # --- OUTPUT TESTS ---

    def test_output_contains_required_lines(self):
        f = io.StringIO()
        with redirect_stdout(f):
            importlib.reload(user_code)
        out = self.normalize_text(f.getvalue())

        _dynamic_test(
            self,
            "restocking started" in out,
            "Output includes 'Restocking started'",
            "Expected 'Restocking started' in output"
        )

        for name in ["bread", "eggs", "milk", "apples"]:
            _dynamic_test(
                self,
                f"restocking {name}" in out,
                f"Output includes 'Restocking {name.title()}'",
                f"Expected 'Restocking {name.title()}' in output"
            )

        _dynamic_test(
            self,
            "restocking completed" in out,
            "Output includes 'Restocking completed'",
            "Expected 'Restocking completed' in output"
        )

    # --- LOOP CHECKS ---

    def test_for_loop_used(self):
        source = strip_comments_from_source(user_code)
        _dynamic_test(
            self,
            re.search(r'\\bfor\\b', source) is not None,
            "'for' loop is used",
            "Expected a 'for' loop"
        )

    def test_while_loop_used(self):
        source = strip_comments_from_source(user_code)
        _dynamic_test(
            self,
            re.search(r'\\bwhile\\b', source) is not None,
            "'while' loop is used",
            "Expected a 'while' loop"
        )


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


test_main.py

Python é uma linguagem de programação de alto nível, interpretada e de uso geral. Diferenciando-se de linguagens como HTML, CSS e JavaScript, que são utilizadas principalmente no desenvolvimento web, Python destaca-se por sua versatilidade em diversos domínios, incluindo desenvolvimento de software, ciência de dados e desenvolvimento back-end. Este curso irá guiá-lo pelos conceitos fundamentais de Python, capacitando-o a criar suas próprias funções ao final do programa.

Nesta seção, você explorará os fundamentos do Python, se familiarizará com o ambiente de trabalho, realizará operações matemáticas simples e exibirá os resultados.

Nesta seção, você descobrirá como armazenar itens como números e textos na memória do computador e recuperá-los pelo nome. Além disso, você terá uma introdução à manipulação de texto em Python.

Nesta seção, você explorará o tipo de dado booleano e aprenderá como ele é utilizado para desenvolver programas que reagem a determinadas condições.

Nesta seção, você continuará aprendendo sobre tipos de dados e descobrirá
como organizar vários atributos em uma única variável utilizando listas, tuplas e dicionários.

Nesta seção, você aprenderá como criar laços de repetição em Python — uma habilidade essencial para iterar por listas, tuplas e dicionários.

Para concluir este curso introdutório, você explorará algumas funções embutidas
e começará a desenvolver as suas próprias!

Desafio: Automatizando o Controle de Estoque

Solução