From 956d76578634a86ea50246de45add748b96cb4ec Mon Sep 17 00:00:00 2001
From: Damien George
Date: Wed, 22 Apr 2015 16:51:29 +0100
Subject: py: Fix printing of "inf" and "nan" floating point values.

---
 py/objfloat.c | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'py/objfloat.c')

diff --git a/py/objfloat.c b/py/objfloat.c
index eb59cc5b7..f74e12f9d 100644
--- a/py/objfloat.c
+++ b/py/objfloat.c
@@ -49,16 +49,16 @@ STATIC void float_print(const mp_print_t *print, mp_obj_t o_in, mp_print_kind_t
     char buf[16];
     mp_format_float(o->value, buf, sizeof(buf), 'g', 7, '\0');
     mp_print_str(print, buf);
-    if (strchr(buf, '.') == NULL && strchr(buf, 'e') == NULL) {
-        // Python floats always have decimal point
+    if (strchr(buf, '.') == NULL && strchr(buf, 'e') == NULL && strchr(buf, 'n') == NULL) {
+        // Python floats always have decimal point (unless inf or nan)
         mp_print_str(print, ".0");
     }
 #else
     char buf[32];
     sprintf(buf, "%.16g", (double) o->value);
     mp_print_str(print, buf);
-    if (strchr(buf, '.') == NULL && strchr(buf, 'e') == NULL) {
-        // Python floats always have decimal point
+    if (strchr(buf, '.') == NULL && strchr(buf, 'e') == NULL && strchr(buf, 'n') == NULL) {
+        // Python floats always have decimal point (unless inf or nan)
         mp_print_str(print, ".0");
     }
 #endif
-- 
cgit v1.2.3