feat: 酷MA萌

src/yuheng/__init__.py

@@ -27,6 +27,26 @@
 from .method.transform import prefix_normalization
 
 
+# 检查模块名称是否被更改
+print(globals().get("__name__", ""))
+
+
+def _check_alias():
+    import sys
+
+    # 检查所有已导入的模块，看是否有模块的名称不是 'yuheng' 但其实是这个包
+    for name, module in sys.modules.items():
+        if (
+            module is not None
+            and name != "yuheng"
+            and getattr(module, "__package__", "") == "yuheng"
+        ):
+            raise ImportError("请不要使用别名导入 yuheng 包。直接使用 import yuheng。")
+
+
+_check_alias()
+
+
 class Carto:
     def __init__(self):
         self.version: str = "0.6"

tests/cases/import_namespace.py

@@ -0,0 +1,30 @@
+import os
+import sys
+import unittest
+
+current_dir = os.path.dirname(os.path.realpath(__file__))
+src_dir = os.path.join(current_dir, "../../src")
+sys.path.append(src_dir)
+
+
+class TestImportNamespace(unittest.TestCase):
+    def setUp(self) -> None:
+        pass
+
+    def test_import_correct(self):
+        import yuheng
+
+        print(globals().get("__name__", ""))
+        world = yuheng.Carto()
+        del world
+
+    def test_import_wrong(self):
+        import yuheng as yh
+
+        print(globals().get("__name__", ""))
+        world = yh.Carto()
+        del world
+
+
+if __name__ == "__main__":
+    unittest.main()