summaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
-rw-r--r--py/builtinimport.c2
-rw-r--r--py/runtime.c4
-rw-r--r--tests/basics/import-pkg4.py2
-rw-r--r--tests/basics/pkg2/__init__.py1
-rw-r--r--tests/basics/pkg2/mod1.py1
-rw-r--r--tests/basics/pkg2/mod2.py1
-rw-r--r--unix/gccollect.c2
7 files changed, 9 insertions, 4 deletions
diff --git a/py/builtinimport.c b/py/builtinimport.c
index 501ced7644..9eede65cd9 100644
--- a/py/builtinimport.c
+++ b/py/builtinimport.c
@@ -218,6 +218,7 @@ mp_obj_t mp_builtin___import__(uint n_args, mp_obj_t *args) {
if (stat == MP_IMPORT_STAT_DIR) {
DEBUG_printf("%s is dir\n", vstr_str(&path));
+ mp_store_attr(module_obj, MP_QSTR___path__, mp_obj_new_str((byte*)vstr_str(&path), vstr_len(&path), false));
vstr_add_char(&path, PATH_SEP_CHAR);
vstr_add_str(&path, "__init__.py");
if (mp_import_stat(vstr_str(&path)) != MP_IMPORT_STAT_FILE) {
@@ -230,7 +231,6 @@ mp_obj_t mp_builtin___import__(uint n_args, mp_obj_t *args) {
vstr_cut_tail_bytes(&path, sizeof("/__init__.py") - 1); // cut off /__init__.py
// https://docs.python.org/3.3/reference/import.html
// "Specifically, any module that contains a __path__ attribute is considered a package."
- mp_store_attr(module_obj, MP_QSTR___path__, mp_obj_new_str((byte*)vstr_str(&path), vstr_len(&path), false));
} else { // MP_IMPORT_STAT_FILE
do_load(module_obj, &path);
// TODO: We cannot just break here, at the very least, we must execute
diff --git a/py/runtime.c b/py/runtime.c
index 4793f054a8..b83cdb13bc 100644
--- a/py/runtime.c
+++ b/py/runtime.c
@@ -1050,7 +1050,7 @@ mp_obj_t mp_import_from(mp_obj_t module, qstr name) {
if (dest[1] != MP_OBJ_NULL) {
// Hopefully we can't import bound method from an object
import_error:
- nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ImportError, "Cannot import name '%s'", qstr_str(name)));
+ nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ImportError, "cannot import name %s", qstr_str(name)));
}
if (dest[0] != MP_OBJ_NULL) {
@@ -1078,7 +1078,7 @@ import_error:
args[1] = mp_const_none; // TODO should be globals
args[2] = mp_const_none; // TODO should be locals
args[3] = mp_const_true; // Pass sentinel "non empty" value to force returning of leaf module
- args[4] = 0;
+ args[4] = MP_OBJ_NEW_SMALL_INT(0);
// TODO lookup __import__ and call that instead of going straight to builtin implementation
return mp_builtin___import__(5, args);
diff --git a/tests/basics/import-pkg4.py b/tests/basics/import-pkg4.py
new file mode 100644
index 0000000000..90b6f2e0ee
--- /dev/null
+++ b/tests/basics/import-pkg4.py
@@ -0,0 +1,2 @@
+# Testing that "recursive" imports (pkg2/__init__.py imports from pkg2) work
+import pkg2
diff --git a/tests/basics/pkg2/__init__.py b/tests/basics/pkg2/__init__.py
new file mode 100644
index 0000000000..101ac7d400
--- /dev/null
+++ b/tests/basics/pkg2/__init__.py
@@ -0,0 +1 @@
+from pkg2 import mod1
diff --git a/tests/basics/pkg2/mod1.py b/tests/basics/pkg2/mod1.py
new file mode 100644
index 0000000000..03754a45f6
--- /dev/null
+++ b/tests/basics/pkg2/mod1.py
@@ -0,0 +1 @@
+from pkg2 import mod2
diff --git a/tests/basics/pkg2/mod2.py b/tests/basics/pkg2/mod2.py
new file mode 100644
index 0000000000..97dadcde46
--- /dev/null
+++ b/tests/basics/pkg2/mod2.py
@@ -0,0 +1 @@
+print("in mod2")
diff --git a/unix/gccollect.c b/unix/gccollect.c
index 33748d240e..35b775bc36 100644
--- a/unix/gccollect.c
+++ b/unix/gccollect.c
@@ -77,7 +77,7 @@ void gc_collect(void) {
//gc_dump_info();
gc_collect_start();
- // this traces .data and .bss sections
+ // this traces the .bss section
#ifdef __CYGWIN__
#define BSS_START __bss_start__
#else