From 6b8b56f8596d3dde544a72d037e9feddd9f70938 Mon Sep 17 00:00:00 2001
From: Damien George <damien.p.george@gmail.com>
Date: Tue, 4 Jul 2017 02:15:11 +1000
Subject: py/modmath: Check for zero division in log with 2 args.

---
 py/modmath.c | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'py')

diff --git a/py/modmath.c b/py/modmath.c
index 10713234c7..d5d135fc11 100644
--- a/py/modmath.c
+++ b/py/modmath.c
@@ -168,6 +168,8 @@ STATIC mp_obj_t mp_math_log(size_t n_args, const mp_obj_t *args) {
         mp_float_t base = mp_obj_get_float(args[1]);
         if (base <= (mp_float_t)0.0) {
             math_error();
+        } else if (base == (mp_float_t)1.0) {
+            mp_raise_msg(&mp_type_ZeroDivisionError, "division by zero");
         }
         return mp_obj_new_float(l / MICROPY_FLOAT_C_FUN(log)(base));
     }
-- 
cgit v1.2.3