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 ""

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

    def test_for_loop_used(self):
        source = strip_comments_from_source(user_code)
        _dynamic_test(
            self,
            re.search(r'for', source),
            "'for' loop is used in the code",
            "Expected use of 'for' loop in the code"
        )

class TestInventoryEvaluation(unittest.TestCase):
    
    def test_if_statements_used(self):
        source = strip_comments_from_source(user_code)
        count = len(re.findall(r'if', source))
        _dynamic_test(
            self,
            count >= 1,
            "'if' condition is used in the code",
            "Expected at least one 'if' condition in the code"
        )



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

    def get_output_lines(self):
        f = io.StringIO()

        with redirect_stdout(f):
            importlib.reload(user_code)

        return [
            self.normalize_text(line)
            for line in f.getvalue().splitlines()
            if line.strip()
        ]

    def test_bread_result(self):
        output = self.get_output_lines()
        expected = self.normalize_text(
            "Bread should be sold at the regular price of 1.2."
        )

        _dynamic_test(
            self,
            expected in output,
            "Bread is sold at the regular price",
            "Incorrect result for Bread"
        )

    def test_eggs_result(self):
        output = self.get_output_lines()
        expected = self.normalize_text(
            "Eggs should be sold at the discounted price of 1.99."
        )

        _dynamic_test(
            self,
            expected in output,
            "Eggs are sold at the discounted price",
            "Incorrect result for Eggs"
        )

    def test_apples_result(self):
        output = self.get_output_lines()
        expected = self.normalize_text(
            "Apples need restocking."
        )

        _dynamic_test(
            self,
            expected in output,
            "Apples need restocking",
            "Incorrect result for Apples"
        )


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

test_main.py

Python is a high-level, interpreted, general-purpose programming
language. Distinguished from languages such as HTML, CSS, and JavaScript, which are mainly
utilized in web development, Python boasts versatility across multiple domains, including
software development, data science, and back-end development. This course will guide you
through Python's fundamental concepts, equipping you with the skills to create your own
functions by the conclusion of the program.

In this section, you will explore the fundamentals of Python, get acquainted
with your workspace, perform simple math operations, and display your findings.

In this section, you'll discover how to save items such as numbers and text
in computer memory and retrieve them by name. Additionally, you'll gain an introduction to text
manipulation in Python.

In this section, you'll explore the boolean data type and learn how it is used
to develop programs that react to certain conditions.

In this section, you will continue learning about data types and discover
how to organize various attributes in a single variable using lists, tuples, and dictionaries.

In this section, you'll learn how to create loops in Python — an essential skill
for iterating through lists, tuples, and dictionaries.

To conclude this introductory course, you will explore some built-in
functions and then begin to develop your own!

Challenge: Solo Coding Capstone

Required Output

Solution