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

class TestTask(unittest.TestCase):
    def test_book_class_exists(self):
        import user_code
        importlib.reload(user_code)
        Book = getattr(user_code, 'Book', None)
        _dynamic_test(
            self,
            Book is not None and isinstance(Book, type),
            "Book class is defined correctly.",
            "Book class is not defined or is not a class."
        )

    def test_book_init_attributes(self):
        import user_code
        importlib.reload(user_code)
        Book = getattr(user_code, 'Book', None)
        _dynamic_test(
            self,
            Book is not None and isinstance(Book, type),
            "Book class is defined for attribute test.",
            "Book class missing for attribute test."
        )
        b = Book("Test Title", "Test Author")
        _dynamic_test(
            self,
            hasattr(b, 'title') and hasattr(b, 'author'),
            "Book object has 'title' and 'author' attributes.",
            "Book object missing 'title' or 'author' attributes."
        )
        _dynamic_test(
            self,
            b.title == "Test Title" and b.author == "Test Author",
            "Book attributes initialized correctly.",
            f"Expected 'Test Title' and 'Test Author', got '{b.title}' and '{b.author}'."
        )

    def test_book1_and_book2_exist_and_different(self):
        import user_code
        importlib.reload(user_code)
        book1 = getattr(user_code, 'book1', None)
        book2 = getattr(user_code, 'book2', None)
        Book = getattr(user_code, 'Book', None)
        _dynamic_test(
            self,
            book1 is not None and book2 is not None,
            "Both book1 and book2 are defined.",
            "book1 and/or book2 are not defined."
        )
        _dynamic_test(
            self,
            isinstance(book1, Book) and isinstance(book2, Book),
            "book1 and book2 are instances of Book.",
            "book1 and/or book2 are not instances of Book."
        )
        _dynamic_test(
            self,
            book1.title != book2.title and book1.author != book2.author,
            "book1 and book2 have different titles and authors.",
            "book1 and book2 must have different titles and authors."
        )

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()


Challenge: Using init and self in Classes

Solução