|
2 | 2 | from datetime import datetime, timedelta |
3 | 3 | from typing import Generator |
4 | 4 |
|
| 5 | +import pandas as pd |
5 | 6 | import pytest |
6 | | -from pytest import raises |
7 | 7 |
|
8 | 8 | from tagreader.clients import IMSClient, list_sources |
9 | 9 | from tagreader.utils import IMSType |
@@ -137,18 +137,17 @@ def test_query_sql(client: IMSClient) -> None: |
137 | 137 | # The % causes WC_E_SYNTAX error in result. Tried "everything" but no go. |
138 | 138 | # Leaving it for now. |
139 | 139 | # query = "SELECT name, ip_description FROM ip_analogdef WHERE name LIKE 'ATC%'" |
140 | | - query = "Select name, ip_description from ip_analogdef where name = 'atcai'" |
| 140 | + query = "Select name, ip_description from ip_analogdef where name = 'atc'" |
141 | 141 | res = client.query_sql(query=query, parse=False) |
142 | | - print(res) |
| 142 | + # print(res) |
143 | 143 | assert isinstance(res, str) |
144 | | - with raises(NotImplementedError): |
145 | | - res = client.query_sql(query=query, parse=True) |
146 | | - assert isinstance(res, str) |
147 | | - client.handler.initialize_connection_string() |
148 | | - query = "Select name, ip_description from ip_analogdef where name = 'atcai'" |
149 | | - res = client.query_sql(query=query, parse=False) |
150 | | - print(res) |
151 | | - assert isinstance(res, str) |
152 | | - with raises(NotImplementedError): |
153 | | - res = client.query_sql(query=query, parse=True) |
154 | | - assert isinstance(res, str) |
| 144 | + |
| 145 | + res = client.query_sql(query=query, parse=True) |
| 146 | + assert isinstance(res, pd.DataFrame) |
| 147 | + assert res.empty |
| 148 | + |
| 149 | + query = "Select name, ip_description from ip_analogdef where name = 'AverageCPUTimeVals'" |
| 150 | + res = client.query_sql(query=query, parse=True) |
| 151 | + assert isinstance(res, pd.DataFrame) |
| 152 | + assert len(res.index.values) == 1 |
| 153 | + assert res.index.values[0] == 0 |
0 commit comments