diff options
Diffstat (limited to 'tests/basics/import_instance_method.py')
| -rw-r--r-- | tests/basics/import_instance_method.py | 38 | 
1 files changed, 38 insertions, 0 deletions
diff --git a/tests/basics/import_instance_method.py b/tests/basics/import_instance_method.py new file mode 100644 index 000000000..d25b70ac5 --- /dev/null +++ b/tests/basics/import_instance_method.py @@ -0,0 +1,38 @@ +# Test importing a method from a class instance. +# This is not a common thing to do, but ensures MicroPython has the same semantics as CPython. + +import sys + +if not hasattr(sys, "modules"): +    print("SKIP") +    raise SystemExit + + +class A: +    def __init__(self, value): +        self.value = value + +    def meth(self): +        return self.value + +    def meth_with_arg(self, a): +        return [self.value, a] + + +# Register a class instance as the module "mod". +sys.modules["mod"] = A(1) + +# Try importing it as a module. +import mod + +print(mod.meth()) +print(mod.meth_with_arg(2)) + +# Change the module. +sys.modules["mod"] = A(3) + +# Try importing it using "from ... import". +from mod import meth, meth_with_arg + +print(meth()) +print(meth_with_arg(4))  | 
