Fix #140 - ToString[] should respect Form.

corywalker · May 29, 2018 · 7abaeb2 · 7abaeb2
1 parent b75227c
commit 7abaeb2
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/expreduce/ex_expression.go b/expreduce/ex_expression.go
@@ -435,6 +435,15 @@ func (this *Expression) StringForm(params ToStringParams) string {
 		}
 	}
 
+	if len(this.Parts) == 2 && isHeadSym && (
+		headAsSym.Name == "System`InputForm" ||
+		headAsSym.Name == "System`FullForm" ||
+		headAsSym.Name == "System`OutputForm") {
+		mutatedParams := params
+		mutatedParams.form = headAsSym.Name[7:]
+		return this.Parts[1].StringForm(mutatedParams)
+	}
+
 	// Default printing format
 	var buffer bytes.Buffer
 	buffer.WriteString(this.Parts[0].StringForm(params))