Added test for multiple statements per query

Alexander Mancevice · Alexander Mancevice · commit e0384b9b15a6 · 2016-02-04T15:39:38.000-05:00
diff --git a/influxdb/tests/dataframe_client_test.py b/influxdb/tests/dataframe_client_test.py
@@ -244,6 +244,56 @@ def test_query_into_dataframe(self):
             for k in expected:
                 assert_frame_equal(expected[k], result[k])
 
+    def test_multiquery_into_dataframe(self):
+        data = {
+            "results": [
+                {
+                    "series": [
+                        {
+                            "name": "cpu_load_short",
+                            "columns": ["time","value"],
+                            "values": [
+                                ["2015-01-29T21:55:43.702900257Z", 0.55],
+                                ["2015-01-29T21:55:43.702900257Z", 23422],
+                                ["2015-06-11T20:46:02Z", 0.64]
+                            ]
+                        }
+                    ]
+                }, {
+                    "series": [
+                        {
+                            "name": "cpu_load_short",
+                            "columns": ["time","count"],
+                            "values": [
+                                ["1970-01-01T00:00:00Z", 3]
+                            ]
+                        }
+                    ]
+                }
+            ]
+        }
+
+        pd1 = pd.DataFrame(
+            [[0.55], [23422.0], [0.64]], columns=['value'],
+            index=pd.to_datetime([
+                "2015-01-29 21:55:43.702900257+0000",
+                "2015-01-29 21:55:43.702900257+0000",
+                "2015-06-11 20:46:02+0000"])).tz_localize('UTC')
+        pd2 = pd.DataFrame(
+            [[3]], columns=['count'],
+            index=pd.to_datetime(["1970-01-01 00:00:00+00:00"])).tz_localize('UTC')
+        expected = [{'cpu_load_short':pd1}, {'cpu_load_short':pd2}]
+
+
+        cli = DataFrameClient('host', 8086, 'username', 'password', 'db')
+        with _mocked_session(cli, 'GET', 200, data):
+            result = cli.query(
+                "SELECT value FROM cpu_load_short WHERE region='us-west';"\
+                "SELECT count(value) FROM cpu_load_short WHERE region='us-west'")
+            for r,e in zip(result, expected):
+                for k in e:
+                    assert_frame_equal(e[k], r[k])
+
     def test_query_with_empty_result(self):
         cli = DataFrameClient('host', 8086, 'username', 'password', 'db')
         with _mocked_session(cli, 'GET', 200, {"results": [{}]}):
