Add unit test for encode_as_numpy.

theengineear · theengineear · commit 111872e2f588 · 2014-12-15T10:08:13.000-08:00
diff --git a/plotly/tests/test_optional/test_utils/test_utils.py b/plotly/tests/test_optional/test_utils/test_utils.py
@@ -4,6 +4,7 @@
 
 from unittest import TestCase
 
+import math
 from datetime import datetime as dt
 import datetime
 import numpy as np
@@ -75,6 +76,19 @@ def test_encode_as_pandas(self):
         res = utils.PlotlyJSONEncoder.encode_as_pandas(pd.NaT)
         self.assertIs(res, None)
 
+    def test_encode_as_numpy(self):
+
+        # should *fail* on non-numpy-y things
+        not_numpy = ['hippo', 8, float('nan'), {'a': 'dict'}]
+        for obj in not_numpy:
+            self.assertRaises(utils.NotEncodable,
+                              utils.PlotlyJSONEncoder.encode_as_numpy, obj)
+
+        # should succeed with numpy-y-thingies
+        res = utils.PlotlyJSONEncoder.encode_as_numpy(np.ma.core.masked)
+        self.assertTrue(math.isnan(res))
+
+
 ## JSON encoding
 numeric_list = [1, 2, 3]
 np_list = np.array([1, 2, 3, np.NaN, np.NAN, np.Inf, dt(2014, 1, 5)])
