#!/usr/bin/env python3

"""

Test script to extract and test Python code execution from Quarto files.

"""

import re

import sys

def extract_and_test_multiprocessing():

"""Extract and test key functions from multiprocessing chapter."""

print("Testing multiprocessing chapter execution...")

try:

with open('en/book/11-multiprocessing.qmd') as f:

content = f.read()

# Test the specific functions we know should work

test_functions = {

'square_number': 'square_number(5)',

'word_count_mapper': 'word_count_mapper("hello world hello")',

'word_count_reducer': 'word_count_reducer({"hello": 2}, {"world": 1})',

'square_mapper': 'square_mapper(5)',

'sum_reducer': 'sum_reducer(10, 20)',

}

# Find and extract all code blocks

pattern = r'```\{python\}(.*?)```'

blocks = re.findall(pattern, content, re.DOTALL)

print(f"Found {len(blocks)} Python code blocks")

# Create a test environment

test_env = {}

# Execute blocks that define functions (skip execution blocks)

for i, block in enumerate(blocks, 1):

block_code = block.strip()

if not block_code:

continue

# Skip blocks that have problematic execution patterns

skip_patterns = [

'pool.map(',

'ProcessPoolExecutor',

'executor.map',

'mp.Pool(',

'multiprocessing.Pool('

]

if any(skip in block_code for skip in skip_patterns):

print(f" Block {i}: SKIPPED (multiprocessing execution)")

continue

# For blocks with if __name__ == "__main__", extract function definitions

if 'if __name__ == "__main__"' in block_code:

# Split and take only the part before if __name__

main_split = block_code.split('if __name__ == "__main__"')[0]

block_code = main_split.strip()

try:

exec(block_code, test_env)

print(f" Block {i}: EXECUTED")

except Exception as e:

print(f" Block {i}: ERROR - {e}")

# Don't fail on execution errors for complex blocks

continue

# Test specific functions

print("\nTesting specific functions:")

for func_name, test_call in test_functions.items():

if func_name in test_env:

try:

result = eval(test_call, test_env)

print(f" {func_name}: OK (result: {result})")

except Exception as e:

print(f" {func_name}: ERROR - {e}")

else:

print(f" {func_name}: NOT FOUND")

print("\nMultiprocessing chapter code validation completed!")

return True

except Exception as e:

print(f"Error: {e}")

return False

if __name__ == "__main__":

success = extract_and_test_multiprocessing()

sys.exit(0 if success else 1)