Test extract_code_from_function()