Test covering multiple time series

timtroendle · timtroendle · commit 2fcf3fa7a929 · 2015-03-14T11:50:30.000+01:00
diff --git a/tests/influxdb/dataframe_client_test.py b/tests/influxdb/dataframe_client_test.py
@@ -203,6 +203,46 @@ def test_query_into_dataframe(self):
             result = cli.query('select column_one from foo;')
             assert_frame_equal(dataframe, result)
 
+    def test_query_multiple_time_series(self):
+        data = [
+            {
+                "name": "series1",
+                "columns": ["time", "mean", "min", "max", "stddev"],
+                "points": [[0, 323048, 323048, 323048, 0]]
+            },
+            {
+                "name": "series2",
+                "columns": ["time", "mean", "min", "max", "stddev"],
+                "points": [[0, -2.8233, -2.8503, -2.7832, 0.0173]]
+            },
+            {
+                "name": "series3",
+                "columns": ["time", "mean", "min", "max", "stddev"],
+                "points": [[0, -0.01220, -0.01220, -0.01220, 0]]
+            }
+        ]
+        dataframes = {
+            'series1': pd.DataFrame(data=[[323048, 323048, 323048, 0]],
+                                    index=pd.to_datetime([0], unit='s',
+                                                         utc=True),
+                                    columns=['mean', 'min', 'max', 'stddev']),
+            'series2': pd.DataFrame(data=[[-2.8233, -2.8503, -2.7832, 0.0173]],
+                                    index=pd.to_datetime([0], unit='s',
+                                                         utc=True),
+                                    columns=['mean', 'min', 'max', 'stddev']),
+            'series3': pd.DataFrame(data=[[-0.01220, -0.01220, -0.01220, 0]],
+                                    index=pd.to_datetime([0], unit='s',
+                                                         utc=True),
+                                    columns=['mean', 'min', 'max', 'stddev'])
+        }
+        with _mocked_session('get', 200, data):
+            cli = DataFrameClient('host', 8086, 'username', 'password', 'db')
+            result = cli.query("""select mean(value), min(value), max(value),
+                stddev(value) from series1, series2, series3""")
+            assert dataframes.keys() == result.keys()
+            for key in dataframes.keys():
+                assert_frame_equal(dataframes[key], result[key])
+
     def test_query_with_empty_result(self):
         with _mocked_session('get', 200, []):
             cli = DataFrameClient('host', 8086, 'username', 'password', 'db')
