You are managing the inventory for a grocery store, and your task is to ensure that the stock for each product is properly maintained and restocked when necessary. 

If an item's stock level **falls below the minimum required amount**, it needs to be restocked.

Additionally, if the stock level of an item **exceeds a certain threshold**, a discount is applied, and the item is marked as "on sale."

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 TestInventoryDiscountRestockSimplified(unittest.TestCase):

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

    # --- STATE / LOGIC TESTS ---

    def test_bread_stock_and_sale_status(self):
        importlib.reload(user_code)
        item = user_code.inventory["Bread"]
        _dynamic_test(
            self,
            item[0] == 50 and item[3] is False,
            "Bread stock updated to 50 and not on sale",
            f"Expected Bread to have 50 units and not be on sale, got {item}"
        )

    def test_eggs_stock_and_sale_status(self):
        importlib.reload(user_code)
        item = user_code.inventory["Eggs"]
        _dynamic_test(
            self,
            item[0] == 200 and item[3] is True,
            "Eggs stock updated to 200 and on sale",
            f"Expected Eggs to have 200 units and be on sale, got {item}"
        )

    def test_milk_stock_and_sale_status(self):
        importlib.reload(user_code)
        item = user_code.inventory["Milk"]
        _dynamic_test(
            self,
            item[0] == 100 and item[3] is False,
            "Milk stock updated to 100 and not on sale",
            f"Expected Milk to have 100 units and not be on sale, got {item}"
        )

    def test_apples_stock_and_sale_status(self):
        importlib.reload(user_code)
        item = user_code.inventory["Apples"]
        _dynamic_test(
            self,
            item[0] == 60 and item[3] is False,
            "Apples stock updated to 60 and not on sale",
            f"Expected Apples to have 60 units and not be on sale, got {item}"
        )

    # --- OUTPUT TESTS (FLEXIBLE, NO \\b) ---

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

        # At least 3 mentions of "process"
        hits = len(re.findall(r'process', out))
        _dynamic_test(
            self,
            hits >= 3,
            "Output contains multiple mentions of processing",
            "Expected at least 3 mentions of processing"
        )

        # Has a final summary line with "processing"
        _dynamic_test(
            self,
            'processing summary' in out or 'processing completed' in out,
            "Output includes a processing summary line",
            "Expected a processing summary line after the loop"
        )
    # --- LOOP USAGE CHECKS ---

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

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


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


test_main.py

Gain a solid foundation in Python for data analysis. Learn to work with variables, data types, strings, conditional statements, loops, core data structures, and functions using examples relevant to analytical tasks.

Challenge: Automating Stock Control

Рішення