From e0e5bd16c6a879455efcb743bbfb4887f258e1a5 Mon Sep 17 00:00:00 2001 From: annatisch Date: Tue, 17 May 2016 11:03:01 -0700 Subject: [PATCH] Fixed failed test --- .../Python/Python/TemplateModels/MethodTemplateModel.cs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/AutoRest/Generators/Python/Python/TemplateModels/MethodTemplateModel.cs b/AutoRest/Generators/Python/Python/TemplateModels/MethodTemplateModel.cs index f1c93804a1b87..2c3a051d9c93d 100644 --- a/AutoRest/Generators/Python/Python/TemplateModels/MethodTemplateModel.cs +++ b/AutoRest/Generators/Python/Python/TemplateModels/MethodTemplateModel.cs @@ -575,17 +575,17 @@ public string GetDocumentationType(IType type) { result = string.Format(CultureInfo.InvariantCulture, "list of {0}", GetDocumentationType(listType.ElementType)); } - else if (type is EnumType) + else if (enumType != null) { - if (type == ReturnType.Body) + if (enumType == ReturnType.Body) { if (enumType.ModelAsString) result = "str"; else - result = string.Format(CultureInfo.InvariantCulture, ":class:`{0} <{1}.models.{0}>`", type.Name, modelNamespace); + result = string.Format(CultureInfo.InvariantCulture, ":class:`{0} <{1}.models.{0}>`", enumType.Name, modelNamespace); } else - result = string.Format(CultureInfo.InvariantCulture, "str or :class:`{0} <{1}.models.{0}>`", type.Name, modelNamespace); + result = string.Format(CultureInfo.InvariantCulture, "str or :class:`{0} <{1}.models.{0}>`", enumType.Name, modelNamespace); } else if (type is DictionaryType) {