Fix int to float comparison

Author: OddCoincidence

tests/snippets/ints.py

@@ -0,0 +1,9 @@
+assert not 1 > 1.0
+assert not 1 < 1.0
+assert 1 >= 1.0
+assert 1 <= 1.0
+
+assert (1).__gt__(1.0) == NotImplemented
+assert (1).__ge__(1.0) == NotImplemented
+assert (1).__lt__(1.0) == NotImplemented
+assert (1).__le__(1.0) == NotImplemented

vm/src/obj/objint.rs

@@ -134,11 +134,13 @@ fn int_lt(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
     arg_check!(
         vm,
         args,
-        required = [
-            (zelf, Some(vm.ctx.int_type())),
-            (other, Some(vm.ctx.int_type()))
-        ]
+        required = [(zelf, Some(vm.ctx.int_type())), (other, None)]
     );
+
+    if !objtype::isinstance(other, &vm.ctx.int_type()) {
+        return Ok(vm.ctx.not_implemented());
+    }
+
     let zelf = BigInt::from_pyobj(zelf);
     let other = BigInt::from_pyobj(other);
     let result = zelf < other;
@@ -149,11 +151,13 @@ fn int_le(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
     arg_check!(
         vm,
         args,
-        required = [
-            (zelf, Some(vm.ctx.int_type())),
-            (other, Some(vm.ctx.int_type()))
-        ]
+        required = [(zelf, Some(vm.ctx.int_type())), (other, None)]
     );
+
+    if !objtype::isinstance(other, &vm.ctx.int_type()) {
+        return Ok(vm.ctx.not_implemented());
+    }
+
     let zelf = BigInt::from_pyobj(zelf);
     let other = BigInt::from_pyobj(other);
     let result = zelf <= other;
@@ -164,11 +168,13 @@ fn int_gt(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
     arg_check!(
         vm,
         args,
-        required = [
-            (zelf, Some(vm.ctx.int_type())),
-            (other, Some(vm.ctx.int_type()))
-        ]
+        required = [(zelf, Some(vm.ctx.int_type())), (other, None)]
     );
+
+    if !objtype::isinstance(other, &vm.ctx.int_type()) {
+        return Ok(vm.ctx.not_implemented());
+    }
+
     let zelf = BigInt::from_pyobj(zelf);
     let other = BigInt::from_pyobj(other);
     let result = zelf > other;
@@ -179,11 +185,13 @@ fn int_ge(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
     arg_check!(
         vm,
         args,
-        required = [
-            (zelf, Some(vm.ctx.int_type())),
-            (other, Some(vm.ctx.int_type()))
-        ]
+        required = [(zelf, Some(vm.ctx.int_type())), (other, None)]
     );
+
+    if !objtype::isinstance(other, &vm.ctx.int_type()) {
+        return Ok(vm.ctx.not_implemented());
+    }
+
     let zelf = BigInt::from_pyobj(zelf);
     let other = BigInt::from_pyobj(other);
     let result = zelf >= other;