numpy dates, datetime.date tests

chriddyp · chriddyp · commit 2b9dfdf49817 · 2014-12-12T14:39:53.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
@@ -3,6 +3,7 @@
 from nose.plugins.attrib import attr
 
 from datetime import datetime as dt
+import datetime
 import numpy as np
 import json
 import pandas as pd
@@ -77,6 +78,17 @@ def test_figure_json_encoding():
                 dt(2014, 1, 5, 1, 1, 1, 1)]))
 
 
+def test_datetime_json_encoding():
+    j1 = json.dumps(dt_list, cls=utils._plotlyJSONEncoder)
+    assert(j1 == '["2014-01-05", '
+                 '"2014-01-05 01:01:01", '
+                 '"2014-01-05 01:01:01.000001"]')
+    j2 = json.dumps({"x": dt_list}, cls=utils._plotlyJSONEncoder)
+    assert(j2 == '{"x": ["2014-01-05", '
+                 '"2014-01-05 01:01:01", '
+                 '"2014-01-05 01:01:01.000001"]}')
+
+
 def test_pandas_json_encoding():
     j1 = json.dumps(df['col 1'], cls=utils._plotlyJSONEncoder)
     assert(j1 == '[1, 2, 3, "2014-01-05", null, NaN, Infinity]')
@@ -136,3 +148,17 @@ def test_masked_constants_example():
                     cls=utils._plotlyJSONEncoder)
     assert(jy == '[-398.11793026999999, -398.11792966000002, '
                  '-398.11786308000001, NaN]')
+
+
+def test_numpy_dates():
+    a = np.arange(np.datetime64('2011-07-11'), np.datetime64('2011-07-18'))
+    j1 = json.dumps(a, cls=utils._plotlyJSONEncoder)
+    assert(j1 == '["2011-07-11", "2011-07-12", "2011-07-13", '
+                 '"2011-07-14", "2011-07-15", "2011-07-16", '
+                 '"2011-07-17"]')
+
+
+def test_datetime_dot_date():
+    a = [datetime.date(2014, 1, 1), datetime.date(2014, 1, 2)]
+    j1 = json.dumps(a, cls=utils._plotlyJSONEncoder)
+    assert(j1 == '["2014-01-01", "2014-01-02"]')
