summaryrefslogtreecommitdiffstatshomepage
path: root/tests/basics/builtin_dir.py
diff options
context:
space:
mode:
authorDamien George <damien.p.george@gmail.com>2018-02-15 18:12:27 +1100
committerDamien George <damien.p.george@gmail.com>2018-02-19 16:12:44 +1100
commit98647e83c733a8a8d36522f71e677367ca8ddd03 (patch)
tree119c07445b7c26f16fbca195dde274af8cbeba61 /tests/basics/builtin_dir.py
parenta8775aaeb053b268cddd629105f02b014df47f64 (diff)
downloadmicropython-98647e83c733a8a8d36522f71e677367ca8ddd03.tar.gz
micropython-98647e83c733a8a8d36522f71e677367ca8ddd03.zip
py/modbuiltins: Simplify and generalise dir() by probing qstrs.
This patch improves the builtin dir() function by probing the target object with all possible qstrs via mp_load_method_maybe. This is very simple (in terms of implementation), doesn't require recursion, and allows to list all methods of user-defined classes (without duplicates) even if they have multiple inheritance with a common parent. The downside is that it can be slow because it has to iterate through all the qstrs in the system, but the "dir()" function is anyway mostly used for testing frameworks and user introspection of types, so speed is not considered a priority. In addition to providing a more complete implementation of dir(), this patch is simpler than the previous implementation and saves some code space: bare-arm: -80 minimal x86: -80 unix x64: -56 unix nanbox: -48 stm32: -80 cc3200: -80 esp8266: -104 esp32: -64
Diffstat (limited to 'tests/basics/builtin_dir.py')
-rw-r--r--tests/basics/builtin_dir.py19
1 files changed, 19 insertions, 0 deletions
diff --git a/tests/basics/builtin_dir.py b/tests/basics/builtin_dir.py
index 16e7e669e4..c15a76f4c6 100644
--- a/tests/basics/builtin_dir.py
+++ b/tests/basics/builtin_dir.py
@@ -17,3 +17,22 @@ foo = Foo()
print('__init__' in dir(foo))
print('x' in dir(foo))
+# dir of subclass
+class A:
+ def a():
+ pass
+class B(A):
+ def b():
+ pass
+d = dir(B())
+print(d.count('a'), d.count('b'))
+
+# dir of class with multiple bases and a common parent
+class C(A):
+ def c():
+ pass
+class D(B, C):
+ def d():
+ pass
+d = dir(D())
+print(d.count('a'), d.count('b'), d.count('c'), d.count('d'))