Rust-related/RustPython
1
0
Fork 1
mirror of https://github.com/RustPython/RustPython.git synced 2026-06-02 19:39:49 +09:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #177 from BojanKogoj/bojan/objfloat-__lt__

Browse Source 
Added __lt__ for float
This commit is contained in:
Windel Bouwman
2018-10-27 22:23:06 +02:00
committed by GitHub
parent 1d854a8fbb a2889f10e9
commit 57dbfd16af
2 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/snippets/floats.py
View File

@@ -1 +1,8 @@
1 + 1.1
a = 1.2
b = 1.3
c = 1.2
assert a < b
assert a <= b
assert a <= c

16
vm/src/obj/objfloat.rs
View File

@@ -61,6 +61,21 @@ fn float_eq(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
Ok(vm.ctx.new_bool(result))
}
fn float_lt(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
arg_check!(
vm,
args,
required = [
(zelf, Some(vm.ctx.float_type())),
(other, Some(vm.ctx.float_type()))
]
);
let zelf = get_value(zelf);
let other = get_value(other);
let result = zelf < other;
Ok(vm.ctx.new_bool(result))
}
fn float_le(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
arg_check!(
vm,
@@ -187,6 +202,7 @@ fn float_pow(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
pub fn init(context: &PyContext) {
let ref float_type = context.float_type;
float_type.set_attr("__eq__", context.new_rustfunc(float_eq));
float_type.set_attr("__lt__", context.new_rustfunc(float_lt));
float_type.set_attr("__le__", context.new_rustfunc(float_le));
float_type.set_attr("__abs__", context.new_rustfunc(float_abs));
float_type.set_attr("__add__", context.new_rustfunc(float_add));