fix lexing issue (drain num_vec)

1 files changed, 5 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
index d9f6b01..66faaee 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -91,6 +91,11 @@ fn lexer(input: &str) -> Result<Vec<Token>, String> {
             }
         }
     }
+    let parse_num = num_vec.parse::<f64>().ok();
+    if let Some(x) = parse_num {
+        result.push(Token::Num(x));
+        num_vec.clear();
+    }
     Ok(result)
 }