commit cc57bd2663fb742770a56f205488dbb2c757fb8f
author Paul Sokolovsky <pfalcon@users.sourceforge.net> Sun Jan 12 01:55:50 2014 +0200
committer Paul Sokolovsky <pfalcon@users.sourceforge.net> Sun Jan 12 03:33:56 2014 +0200
tree b4951a38f8c1635605c55246348893eb57efe744
parent 729e9cce7bd31d3f107a4d6e9498b0fa27119e22

mp_obj_equal(): For non-trivial types, call out to type's special method.

py/obj.c

diff --git a/py/obj.c b/py/obj.c
index 81b5c69..2759437 100644
--- a/py/obj.c
+++ b/py/obj.c
@@ -117,6 +117,13 @@
     } else if (MP_OBJ_IS_TYPE(o1, &str_type) && MP_OBJ_IS_TYPE(o2, &str_type)) {
         return mp_obj_str_get(o1) == mp_obj_str_get(o2);
     } else {
+        mp_obj_base_t *o = o1;
+        if (o->type->binary_op != NULL) {
+            mp_obj_t r = o->type->binary_op(RT_COMPARE_OP_EQUAL, o1, o2);
+            if (r != MP_OBJ_NULL) {
+                return r == mp_const_true ? true : false;
+            }
+        }
         // TODO: Debugging helper
         printf("Equality for '%s' and '%s' types not yet implemented\n", mp_obj_get_type_str(o1), mp_obj_get_type_str(o2));
         assert(0);