From 1e87b56219c69306d77a887cac3d29146180f113 Mon Sep 17 00:00:00 2001
From: Andrew Leech <andrew.leech@planetinnovation.com.au>
Date: Mon, 4 Jul 2022 17:35:46 +1000
Subject: py/obj: Add support for __float__ and __complex__ functions.

---
 py/objcomplex.c | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'py/objcomplex.c')

diff --git a/py/objcomplex.c b/py/objcomplex.c
index 157617e156..3c4cb66140 100644
--- a/py/objcomplex.c
+++ b/py/objcomplex.c
@@ -88,6 +88,10 @@ STATIC mp_obj_t complex_make_new(const mp_obj_type_t *type_in, size_t n_args, si
                 // a complex, just return it
                 return args[0];
             } else {
+                mp_float_t real, imag;
+                if (mp_obj_get_complex_maybe(args[0], &real, &imag)) {
+                    return mp_obj_new_complex(real, imag);
+                }
                 // something else, try to cast it to a complex
                 return mp_obj_new_complex(mp_obj_get_float(args[0]), 0);
             }
-- 
cgit v1.2.3