test: add inherit.er

crates/erg_compiler/context/inquire.rs

@@ -1271,7 +1271,7 @@ impl Context {
                 let one = attr_name
                     .as_ref()
                     .map(|attr| {
-                        ctxs.into_iter()
+                        ctxs.iter()
                             .flat_map(|ctx| {
                                 ctx.get_singular_ctxs_by_ident(attr, self)
                                     .ok()
@@ -1279,7 +1279,7 @@ impl Context {
                             })
                             .collect()
                     })
-                    .unwrap_or(vec![]);
+                    .unwrap_or(ctxs);
                 let two = obj
                     .qual_name()
                     .map(|name| {

tests/should_ok/inherit.er

@@ -0,0 +1,15 @@
+@Inheritable
+C = Class { .x = Int }
+C.
+    const = 0
+C.
+    new x: Int = C::__new__ { .x = x }
+
+D = Inherit C
+D.
+    @Override
+    new x: Int = D::__new__ { .x = x }
+
+d: D = D.new(1)
+print! d
+assert D.const == 0

tests/test.rs

@@ -81,6 +81,11 @@ fn exec_import() -> Result<(), ()> {
     expect_success("examples/import.er")
 }
 
+#[test]
+fn exec_inherit() -> Result<(), ()> {
+    expect_success("tests/should_ok/inherit.er")
+}
+
 #[test]
 fn exec_infer_class() -> Result<(), ()> {
     expect_success("tests/should_ok/infer_class.er")