fix: ByteArray! is not defined

erg-lang · Jun 15, 2024 · 325ab03 · 325ab03
1 parent 22cd07a
commit 325ab03
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 22 deletions.
diff --git a/crates/erg_compiler/context/initialize/classes.rs b/crates/erg_compiler/context/initialize/classes.rs
@@ -3222,34 +3222,34 @@ impl Context {
         );
         list_mut_.register_trait_methods(list_mut_t.clone(), list_mut_mutable);
         /* ByteArray! */
-        let bytelist_mut_t = mono(BYTEARRAY);
-        let mut bytelist_mut = Self::builtin_mono_class(BYTEARRAY, 2);
+        let bytearray_mut_t = mono(MUT_BYTEARRAY);
+        let mut bytearray_mut = Self::builtin_mono_class(MUT_BYTEARRAY, 2);
         let t_append = pr_met(
-            ref_mut(bytelist_mut_t.clone(), None),
+            ref_mut(bytearray_mut_t.clone(), None),
             vec![kw(KW_ELEM, int_interval(IntervalOp::Closed, 0, 255))],
             None,
             vec![],
             NoneType,
         );
-        bytelist_mut.register_builtin_py_impl(
+        bytearray_mut.register_builtin_py_impl(
             PROC_PUSH,
             t_append,
             Immutable,
             Visibility::BUILTIN_PUBLIC,
             Some(FUNC_APPEND),
         );
-        let t_copy = fn0_met(ref_(bytelist_mut_t.clone()), bytelist_mut_t.clone());
-        let mut bytelist_mut_copy = Self::builtin_methods(Some(mono(COPY)), 2);
-        bytelist_mut_copy.register_builtin_py_impl(
+        let t_copy = fn0_met(ref_(bytearray_mut_t.clone()), bytearray_mut_t.clone());
+        let mut bytearray_mut_copy = Self::builtin_methods(Some(mono(COPY)), 2);
+        bytearray_mut_copy.register_builtin_py_impl(
             FUNC_COPY,
             t_copy,
             Immutable,
             Visibility::BUILTIN_PUBLIC,
             Some(FUNC_COPY),
         );
-        bytelist_mut.register_trait_methods(bytelist_mut_t.clone(), bytelist_mut_copy);
+        bytearray_mut.register_trait_methods(bytearray_mut_t.clone(), bytearray_mut_copy);
         let t_extend = pr_met(
-            ref_mut(bytelist_mut_t.clone(), None),
+            ref_mut(bytearray_mut_t.clone(), None),
             vec![kw(
                 KW_ITERABLE,
                 poly(
@@ -3261,15 +3261,15 @@ impl Context {
             vec![],
             NoneType,
         );
-        bytelist_mut.register_builtin_py_impl(
+        bytearray_mut.register_builtin_py_impl(
             PROC_EXTEND,
             t_extend,
             Immutable,
             Visibility::BUILTIN_PUBLIC,
             Some(FUNC_EXTEND),
         );
         let t_insert = pr_met(
-            ref_mut(bytelist_mut_t.clone(), None),
+            ref_mut(bytearray_mut_t.clone(), None),
             vec![
                 kw(KW_INDEX, Nat),
                 kw(KW_ELEM, int_interval(IntervalOp::Closed, 0, 255)),
@@ -3278,26 +3278,26 @@ impl Context {
             vec![],
             NoneType,
         );
-        bytelist_mut.register_builtin_py_impl(
+        bytearray_mut.register_builtin_py_impl(
             PROC_INSERT,
             t_insert,
             Immutable,
             Visibility::BUILTIN_PUBLIC,
             Some(FUNC_INSERT),
         );
         let t_pop = pr0_met(
-            ref_mut(bytelist_mut_t.clone(), None),
+            ref_mut(bytearray_mut_t.clone(), None),
             int_interval(IntervalOp::Closed, 0, 255),
         );
-        bytelist_mut.register_builtin_py_impl(
+        bytearray_mut.register_builtin_py_impl(
             PROC_POP,
             t_pop,
             Immutable,
             Visibility::BUILTIN_PUBLIC,
             Some(FUNC_POP),
         );
-        let t_reverse = pr0_met(ref_mut(bytelist_mut_t.clone(), None), NoneType);
-        bytelist_mut.register_builtin_py_impl(
+        let t_reverse = pr0_met(ref_mut(bytearray_mut_t.clone(), None), NoneType);
+        bytearray_mut.register_builtin_py_impl(
             PROC_REVERSE,
             t_reverse,
             Immutable,
@@ -3931,11 +3931,11 @@ impl Context {
         self.register_builtin_type(mono(MUT_FILE), file_mut, vis.clone(), Const, Some(FILE));
         self.register_builtin_type(list_mut_t, list_mut_, vis.clone(), Const, Some(LIST));
         self.register_builtin_type(
-            bytelist_mut_t,
-            bytelist_mut,
+            bytearray_mut_t,
+            bytearray_mut,
             vis.clone(),
             Const,
-            Some(BYTEARRAY),
+            Some(FUNC_BYTEARRAY),
         );
         self.register_builtin_type(dict_mut_t, dict_mut, vis.clone(), Const, Some(DICT));
         self.register_builtin_type(set_mut_t, set_mut_, vis.clone(), Const, Some(SET));

diff --git a/crates/erg_compiler/context/initialize/funcs.rs b/crates/erg_compiler/context/initialize/funcs.rs
@@ -70,7 +70,7 @@ impl Context {
         let t_bytes_array = no_var_func(
             vec![],
             vec![kw(KW_ITERABLE, poly(ITERABLE, vec![ty_tp(Int)]))],
-            mono(BYTEARRAY),
+            mono(MUT_BYTEARRAY),
         );
         let t_callable = func1(Obj, Bool);
         let t_chr = nd_func(
@@ -275,7 +275,7 @@ impl Context {
         let t_memoryview = nd_func(
             vec![kw(
                 KW_OBJ,
-                mono(BYTES) | mono(BYTEARRAY) | mono("array.Array!"),
+                mono(BYTES) | mono(MUT_BYTEARRAY) | mono("array.Array!"),
             )],
             None,
             mono(MEMORYVIEW),

diff --git a/crates/erg_compiler/context/initialize/mod.rs b/crates/erg_compiler/context/initialize/mod.rs
@@ -141,7 +141,7 @@ const NEVER: &str = "Never";
 const OBJ: &str = "Obj";
 const MUTABLE_OBJ: &str = "Obj!";
 const BYTES: &str = "Bytes";
-const BYTEARRAY: &str = "ByteArray!";
+const MUT_BYTEARRAY: &str = "ByteArray!";
 const FLOAT: &str = "Float";
 const MUT_FLOAT: &str = "Float!";
 const EPSILON: &str = "EPSILON";