masked constants test

Author: chriddyp

plotly/tests/test_optional/test_utils/test_utils.py

@@ -7,16 +7,21 @@
 import json
 import pandas as pd
 from pandas.util.testing import assert_series_equal
+import matplotlib.pyplot as plt
 
 from plotly import utils
 from plotly.grid_objs import Column
 from plotly.graph_objs import Scatter, Scatter3d, Figure, Data
+from plotly.matplotlylib import Exporter, PlotlyRenderer
+from plotly.plotly import plot
 
 ## JSON encoding
 numeric_list = [1, 2, 3]
 np_list = np.array([1, 2, 3, np.NaN, np.NAN, np.Inf, dt(2014, 1, 5)])
 mixed_list = [1, 'A', dt(2014, 1, 5), dt(2014, 1, 5, 1, 1, 1),
               dt(2014, 1, 5, 1, 1, 1, 1)]
+dt_list = [dt(2014, 1, 5), dt(2014, 1, 5, 1, 1, 1),
+           dt(2014, 1, 5, 1, 1, 1, 1)]
 
 df = pd.DataFrame(columns=['col 1'],
                   data=[1, 2, 3, dt(2014, 1, 5), pd.NaT, np.NaN, np.Inf])
@@ -98,3 +103,36 @@ def test_pandas_json_encoding():
 
     j6 = json.dumps(ts.index, cls=utils._plotlyJSONEncoder)
     assert(j6 == '["2011-01-01", "2011-01-01 01:00:00"]')
+
+
+def test_numpy_masked_json_encoding():
+    l = [1, 2, np.ma.core.masked]
+    j1 = json.dumps(l, cls=utils._plotlyJSONEncoder)
+    assert(j1 == '[1, 2, NaN]')
+    assert(set(l) == set([1, 2, np.ma.core.masked]))
+
+
+def test_masked_constants_example():
+    # example from: https://gist.github.com/tschaume/d123d56bf586276adb98
+    data = {
+        'esN': [0, 1, 2, 3],
+        'ewe_is0': [-398.11901997, -398.11902774,
+                    -398.11897111, -398.11882215],
+        'ewe_is1': [-398.11793027, -398.11792966, -398.11786308, None],
+        'ewe_is2': [-398.11397008, -398.11396421, None, None]
+    }
+    df = pd.DataFrame.from_dict(data)
+
+    plotopts = {'x': 'esN', 'marker': 'o'}
+    fig, ax = plt.subplots(1, 1)
+    df.plot(ax=ax, **plotopts)
+
+    renderer = PlotlyRenderer()
+    Exporter(renderer).run(fig)
+
+    json.dumps(renderer.plotly_fig, cls=utils._plotlyJSONEncoder)
+
+    jy = json.dumps(renderer.plotly_fig['data'][1]['y'],
+                    cls=utils._plotlyJSONEncoder)
+    assert(jy == '[-398.11793026999999, -398.11792966000002, '
+                 '-398.11786308000001, NaN]')