__unm, __bnot

2 files changed, 75 insertions, 5 deletions

diff --git a/README.md b/README.md
index 9dce060..9505854 100644
--- a/README.md
+++ b/README.md
@@ -22,7 +22,6 @@
     - [x] `__gc` (unavailable)
     - [x] `__mode` (unavailable)
     - [ ] `__len`
-    - [ ] `__eq`
     - [x] `__add`
     - [x] `__sub`
     - [x] `__mul`
@@ -35,8 +34,9 @@
     - [x] `__bxor`
     - [x] `__shl`
     - [x] `__shr`
-    - [ ] `__unm`
-    - [ ] `__bnot`
+    - [x] `__unm`
+    - [x] `__bnot`
+    - [x] `__eq`
     - [x] `__lt`
     - [x] `__le`
     - [ ] `__concat`
diff --git a/tests/ltm.js b/tests/ltm.js
index b3f5e30..8a20f67 100644
--- a/tests/ltm.js
+++ b/tests/ltm.js
@@ -1005,8 +1005,6 @@ test('__le', function (t) {
 });
 
 
-
-
 test('__le that uses __lt', function (t) {
     let luaCode = `
         local mt = {
@@ -1067,4 +1065,76 @@ test('__le that uses __lt', function (t) {
         true,
         "Program output is correct"
     );
+});
+
+
+test('__unm, __bnot', function (t) {
+    let luaCode = `
+        local mt = {
+            __unm = function (a)
+                return "hello"
+            end,
+
+            __bnot = function (a)
+                return "world"
+            end
+        }
+
+        local t = {}
+
+        -- setmetatable(t, mt)
+
+        return -t, ~t
+    `, L;
+    
+    t.plan(5);
+
+    t.comment("Running following code: \n" + luaCode);
+
+    // main <hello.lua:0,0> (10 instructions at 0x7ff10e403210)
+    // 0+ params, 4 slots, 1 upvalue, 2 locals, 2 constants, 2 functions
+    //         1       [1]     NEWTABLE        0 0 2
+    //         2       [4]     CLOSURE         1 0     ; 0x7ff10e403440
+    //         3       [4]     SETTABLE        0 -1 1  ; "__unm" -
+    //         4       [8]     CLOSURE         1 1     ; 0x7ff10e4033c0
+    //         5       [8]     SETTABLE        0 -2 1  ; "__bnot" -
+    //         6       [11]    NEWTABLE        1 0 0
+    //         7       [15]    UNM             2 1                      <=== We stop here
+    //         8       [15]    BNOT            3 1
+    //         9       [15]    RETURN          2 3
+    //         10      [15]    RETURN          0 1
+    //
+    // ...
+
+    t.doesNotThrow(function () {
+        L = getState(luaCode);
+    }, "Bytecode parsed without errors");
+
+    L.stack[0].p.code[6].breakpoint = true;
+
+    t.doesNotThrow(function () {
+        ldo.luaD_call(L, 0, -1);
+    }, "First part of the program executed without errors");
+
+    L.ci.pcOff--;
+    L.stack[0].p.code[6].breakpoint = false;
+
+    t.comment("We manually set t's metatable to mt");
+    L.stack[2].metatable = L.stack[1];
+
+    t.doesNotThrow(function () {
+        VM.luaV_execute(L);
+    }, "Second part of the program executed without errors");
+
+    t.strictEqual(
+        L.stack[L.top - 1].value,
+        "world",
+        "Program output is correct"
+    );
+
+    t.strictEqual(
+        L.stack[L.top - 2].value,
+        "hello",
+        "Program output is correct"
+    );
 });
\ No newline at end of file