From b7b158859494bf5f9d8b3c1d4d120eeb200c83c0 Mon Sep 17 00:00:00 2001
From: Shunsuke Shibayama <sbym1346@gmail.com>
Date: Wed, 2 Aug 2023 18:41:28 +0900
Subject: [PATCH] fix: namespace bug

---
 crates/erg_compiler/context/inquire.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/crates/erg_compiler/context/inquire.rs b/crates/erg_compiler/context/inquire.rs
index 2311aa81..a3941a1f 100644
--- a/crates/erg_compiler/context/inquire.rs
+++ b/crates/erg_compiler/context/inquire.rs
@@ -2606,6 +2606,17 @@ impl Context {
     }
 
     pub(crate) fn get_namespace_path(&self, namespace: &Str) -> Option<PathBuf> {
+        // get the true name
+        let namespace = if let Some((_, vi)) = self.get_var_info(namespace) {
+            // m: PyModule("math") -> math
+            if vi.t.is_module() {
+                vi.t.typarams()[0].to_string().replace('"', "").into()
+            } else {
+                namespace.clone()
+            }
+        } else {
+            namespace.clone()
+        };
         let mut namespaces = namespace.split_with(&[".", "::"]);
         let mut str_namespace = namespaces.first().map(|n| n.to_string())?;
         namespaces.remove(0);