fix blanks

blackchip-org · Mar 5, 2024 · da50c77 · da50c77
1 parent 3286d1d
commit da50c77
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/pkg/repl/repl.go b/pkg/repl/repl.go
@@ -106,9 +106,8 @@ func (r *REPL) Eval(line string) error {
 	r.err = nil
 	prev := r.Calc.Stack()
 
-	line = strings.TrimSpace(line)
 	if r.EndQuote != "" {
-		if line == r.EndQuote {
+		if strings.TrimSpace(line) == r.EndQuote {
 			r.EndQuote = ""
 		} else {
 			r.Calc.Push(line)

diff --git a/pkg/repl/repl_test.go b/pkg/repl/repl_test.go
@@ -83,7 +83,7 @@ func TestQuoteBlanks(t *testing.T) {
 	repl.Eval("EOF")
 
 	have := c.Stack()
-	want := []string{"1 2 add", "2 3 sub"}
+	want := []string{"1 2 add", "2 3 sub", "", ""}
 	if !reflect.DeepEqual(have, want) {
 		t.Fatalf("\n have: %v \n want: %v", have, want)
 	}