TimeSeries type v2 (#666)

* TimeSeries -> TimeSeries2 , better inheritance and tests

analytics/analytics/analytic_types/__init__.py

@@ -28,9 +28,12 @@ ts = TimeSeries([4, 5, 6], tsis)
 Timestamp = Union[str, pd.Timestamp]
 
 class TimeSeriesIndex(pd.DatetimeIndex):
-    def __init__(self, timestamps: List[Timestamp]):
+    def __new__(cls, *args, **kwargs):
-        super().__init__(timestamps, dtype='datetime64[ns]')
+        return pd.DatetimeIndex.__new__(cls, *args, **kwargs)
-
+
-class TimeSeries(pd.Series):
+# TODO: make generic type for values. See List definition for example of generic class
-    def __init__(self, values: List[object], tsindex: TimeSeriesIndex):
+# TODO: constructor from DataFrame
-        super().__init__(values, index=tsindex)
+# TODO: repleace TimeSeries (above) with this class: rename TimeSeries2 to TimeSeries
+class TimeSeries2(pd.Series):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)

analytics/tests/test_analytic_types.py

@@ -0,0 +1,16 @@
+from analytic_types import TimeSeriesIndex, TimeSeries2
+
+import unittest
+
+
+class TestDataset(unittest.TestCase):
+    def test_basic_timeseries_index(self):
+        tsi = TimeSeriesIndex(['2017-12-31 16:00:00-08:00'])
+        self.assertEqual(len(tsi), 1)
+        tsi2 = TimeSeriesIndex(['2017-12-31 16:00:00-08:00', '2017-12-31 17:00:00-08:00', '2017-12-31 18:00:00-08:00'])
+        self.assertEqual(len(tsi2), 3)
+
+    def test_basic_timeseries(self):
+        tsis = TimeSeriesIndex(['2017-12-31 16:00:00-08:00', '2017-12-31 17:00:00-08:00', '2017-12-31 18:00:00-08:00'])
+        ts = TimeSeries2([4, 5, 6], tsis)
+        self.assertEqual(len(ts), 3)