Added tests

tarzzz · tarzzz · commit 204a048af99c · 2016-12-30T17:26:36.000+05:30
diff --git a/plotly/tests/test_optional/test_utils/test_utils.py b/plotly/tests/test_optional/test_utils/test_utils.py
@@ -7,6 +7,7 @@
 import datetime
 import json
 import math
+import decimal
 from datetime import datetime as dt
 from unittest import TestCase
 
@@ -149,6 +150,13 @@ def test_encode_as_date(self):
         )
         self.assertEqual(res, '2013-10-01 00:00:00.000010')
 
+    def test_encode_as_decimal(self):
+
+        # should work with decimal values
+        res = utils.PlotlyJSONEncoder.encode_as_decimal(decimal.Decimal(1.023452))
+
+        self.assertAlmostEqual(res, 1.023452) # Checks upto 7 decimal places
+        self.assertIsInstance(res, float)
 
 ## JSON encoding
 numeric_list = [1, 2, 3]
diff --git a/plotly/utils.py b/plotly/utils.py
@@ -131,9 +131,6 @@ def coerce_to_strict(self, const):
         # before python 2.7, 'true', 'false', 'null', were include here.
         if const in ('Infinity', '-Infinity', 'NaN'):
             return None
-        # Decimal Fails on Serialization, convert it to float
-        elif isinstance(const, decimal.Decimal):
-            return float(const)
         else:
             return const
 
