def METHOD_NAME(self):
    error = APIException({
        'non_field_errors': ['Error message.']
    })
    response = exception_handler(error, None)
    assert response.data == {
        'code': 'error',
        'message': 'Error message.',
    }