def METHOD_NAME(custom_extractor, data, expected):
    extractor = custom_extractor()
    response = requests.Response()
    response.json = MagicMock(return_value={'data': data})
    result = extractor.extract_records(response)
    assert result == expected