Fix example

Author: shellyln

_examples/torpn/torpn.go

@@ -155,18 +155,19 @@ func Parse(s string) ([]interface{}, error) {
 	}
 
 	if out.MatchStatus == MatchStatus_Matched {
-		var result AstSlice
 		switch out.AstStack[0].Type {
 		case AstType_Int:
-			result = AstSlice{out.AstStack[0].Value.(Ast)}
+			return []interface{}{out.AstStack[0].Value.(int64)}, nil
 		case AstType_ListOfAst:
-			result = out.AstStack[0].Value.(AstSlice)
-		}
-		ret := make([]interface{}, 0, len(result))
-		for _, v := range result {
-			ret = append(ret, v.Value)
+			result := out.AstStack[0].Value.(AstSlice)
+			ret := make([]interface{}, 0, len(result))
+			for _, v := range result {
+				ret = append(ret, v.Value)
+			}
+			return ret, nil
+		default:
+			return nil, errors.New("Unexpected: no result")
 		}
-		return ret, nil
 	} else {
 		pos := GetLineAndColPosition(s, out.SourcePosition, 4)
 		return nil, errors.New(

_examples/torpn/torpn_test.go

@@ -80,6 +80,11 @@ func TestParse(t *testing.T) {
 		args:    args{s: "(3+5)*(7+11)"},
 		want:    []interface{}{int64(3), int64(5), "+", int64(7), int64(11), "+", "*"},
 		wantErr: false,
+	}, {
+		name:    "8",
+		args:    args{s: "13"},
+		want:    []interface{}{int64(13)},
+		wantErr: false,
 	}}
 
 	runMatrixParse(t, tests)