
import unittest
import user_code
import ast
import re   
import importlib
import csv
import unittest
import importlib

class TestTask(unittest.TestCase):
    def test_unpack_first_last_typical(self):
        import user_code
        importlib.reload(user_code)
        func = getattr(user_code, 'unpack_first_last', None)
        _dynamic_test(
            self,
            callable(func),
            "Function 'unpack_first_last' is defined.",
            "Function 'unpack_first_last' is not defined."
        )
        result = func([1, 2, 3, 4, 5, 6]) if func else None
        expected = (1, 2, 5, 6)
        _dynamic_test(
            self,
            result == expected,
            "Correct tuple for list [1, 2, 3, 4, 5, 6]",
            f"Expected {expected}, got {result}"
        )

    def test_unpack_first_last_strings(self):
        import user_code
        importlib.reload(user_code)
        func = getattr(user_code, 'unpack_first_last', None)
        result = func(['a', 'b', 'c', 'd']) if func else None
        expected = ('a', 'b', 'c', 'd')
        _dynamic_test(
            self,
            result == expected,
            "Correct tuple for list ['a', 'b', 'c', 'd']",
            f"Expected {expected}, got {result}"
        )

    def test_unpack_first_last_five_elements(self):
        import user_code
        importlib.reload(user_code)
        func = getattr(user_code, 'unpack_first_last', None)
        result = func([10, 20, 30, 40, 50]) if func else None
        expected = (10, 20, 40, 50)
        _dynamic_test(
            self,
            result == expected,
            "Correct tuple for list [10, 20, 30, 40, 50]",
            f"Expected {expected}, got {result}"
        )

    def test_unpack_first_last_four_elements(self):
        import user_code
        importlib.reload(user_code)
        func = getattr(user_code, 'unpack_first_last', None)
        result = func([7, 8, 9, 10]) if func else None
        expected = (7, 8, 9, 10)
        _dynamic_test(
            self,
            result == expected,
            "Correct tuple for list [7, 8, 9, 10]",
            f"Expected {expected}, got {result}"
        )

    def test_unpack_first_last_too_short(self):
        import user_code
        importlib.reload(user_code)
        func = getattr(user_code, 'unpack_first_last', None)
        failed = False
        try:
            func([1, 2, 3])
        except Exception:
            failed = True
        _dynamic_test(
            self,
            failed,
            "Function raises exception for list shorter than 4 elements.",
            "Function should raise exception for lists shorter than 4 elements."
        )

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 normalize_text(text):
    text = text.lower()
    text = re.sub(r"\\s{2,}", " ", text)
    text = re.sub(r"\\s*([,:?])\\s*", r"\\1 ", text)
    return text.strip()

def change_var(code: str, var_name: str, value: str) -> str:
    tree = ast.parse(code)
    lines = code.splitlines()
    changed = False
    # Collect all assignment nodes to modify
    assign_nodes = [
        (i, node)
        for i, node in enumerate(tree.body)
        if isinstance(node, ast.Assign)
        and any(isinstance(target, ast.Name) and target.id == var_name for target in node.targets)
    ]

    # If nothing to change, return unmodified code
    if not assign_nodes:
        return code

    # Perform replacements for all matching assignments (from last to first to not break line offsets)
    for i, node in reversed(assign_nodes):
        start_line = node.lineno - 1
        line = lines[start_line]
        indent = ' ' * (len(line) - len(line.lstrip()))
        lines[start_line] = f"{indent}{var_name} = {value}"
        next_line = len(lines)
        for next_node in tree.body[i+1:]:
            if hasattr(next_node, 'lineno'):
                next_line = next_node.lineno - 1
                break
        if next_line > start_line + 1:
            lines[start_line+1:next_line] = []
        changed = True

    return '\\n'.join(lines) if changed else code

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


test_main.py

У цьому курсі ви вивчите такі складні теми, як аргументи функцій, ітератори, генератори, замикання та декоратори. Десятки практичних вправ допоможуть вам засвоїти ці теми.

Explore the art of packing and unpacking lists, tuples, and dictionaries to write more expressive and efficient Python code.

Delve into the world of higher-order functions, passing functions as arguments, and writing concise anonymous functions with lambda.

Unravel the mysteries of variable scope, and learn how nested functions and closures enable powerful encapsulation.

Занурення в протокол ітерації Python, користувацькі ітератори та генераторні функції для ефективної обробки даних.

Challenge: Unpacking Multiple Elements

Рішення