From 3971c2a09ab6fedc6bdcf5b7597fbf3d0bafeb0d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gustav=20S=C3=B6rn=C3=A4s?= <gustav@sornas.net>
Date: Sat, 9 Jan 2021 20:58:40 +0100
Subject: assert operator

---
 src/vm.rs | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

(limited to 'src/vm.rs')

diff --git a/src/vm.rs b/src/vm.rs
index 9102f81..633b920 100644
--- a/src/vm.rs
+++ b/src/vm.rs
@@ -21,10 +21,12 @@ pub enum Op {
     Or,
     Not,
 
-    Equal, // ==
-    Less, // <
+    Equal,   // ==
+    Less,    // <
     Greater, // >
 
+    AssertEqual,
+
     Print,
     Return,
 }
@@ -83,6 +85,10 @@ impl VM {
         (b, a)
     }
 
+    fn _peek_up(&self, amount: usize) -> Option<&Value> {
+        self.stack.get(self.stack.len() - amount)
+    }
+
     pub fn run(&mut self) {
         const PRINT_WHILE_RUNNING: bool = true;
         const PRINT_BLOCK: bool = true;
@@ -192,6 +198,14 @@ impl VM {
                     }
                 }
 
+                Op::AssertEqual => {
+                    let (a, b) = self.pop_twice();
+                    if a != b {
+                        println!("Assert failed for '{:?}' and '{:?}'", a, b);
+                    }
+                    self.stack.push(Value::Bool(a == b));
+                }
+
                 Op::Print => {
                     println!("PRINT: {:?}", self.stack.pop());
                 }
-- 
cgit v1.2.1