language-servers/erg
1
0
Fork 0
mirror of https://github.com/erg-lang/erg.git synced 2025-08-04 10:49:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: modulo operator bug

Browse source
This commit is contained in:
Shunsuke Shibayama 2024-02-04 18:11:30 +09:00
parent 8189d1cfc4
commit 96239b957c
3 changed files with 51 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

42
crates/erg_compiler/context/initialize/classes.rs
View file

@ -569,6 +569,27 @@ impl Context {
ValueObj::builtin_class(Int),
);
int.register_trait(Int, int_floordiv);
// needed for implementing `%` operator
let mut int_div = Self::builtin_methods(Some(poly(DIV, vec![ty_tp(Int)])), 2);
int_div.register_builtin_erg_impl(
OP_DIV,
fn1_met(Int, Int, Float),
Const,
Visibility::BUILTIN_PUBLIC,
);
int_div.register_builtin_const(
OUTPUT,
Visibility::BUILTIN_PUBLIC,
None,
ValueObj::builtin_class(Float),
);
int_div.register_builtin_const(
MOD_OUTPUT,
Visibility::BUILTIN_PUBLIC,
None,
ValueObj::builtin_class(Int),
);
int.register_trait(Int, int_div);
let mut int_pos = Self::builtin_methods(Some(mono(POS)), 2);
int_pos.register_builtin_const(
OUTPUT,
@ -708,6 +729,27 @@ impl Context {
ValueObj::builtin_class(Nat),
);
nat.register_trait(Nat, nat_floordiv);
// needed for implementing `%` operator
let mut nat_div = Self::builtin_methods(Some(poly(DIV, vec![ty_tp(Nat)])), 2);
nat_div.register_builtin_erg_impl(
OP_DIV,
fn1_met(Nat, Nat, Float),
Const,
Visibility::BUILTIN_PUBLIC,
);
nat_div.register_builtin_const(
OUTPUT,
Visibility::BUILTIN_PUBLIC,
None,
ValueObj::builtin_class(Float),
);
nat_div.register_builtin_const(
MOD_OUTPUT,
Visibility::BUILTIN_PUBLIC,
None,
ValueObj::builtin_class(Nat),
);
nat.register_trait(Nat, nat_div);
let mut nat_mutizable = Self::builtin_methods(Some(mono(MUTIZABLE)), 2);
nat_mutizable.register_builtin_const(
MUTABLE_MUT_TYPE,