fix ut

apache · Jan 28, 2022 · 0506ddf · 0506ddf
1 parent bf3dce2
commit 0506ddf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/integration_tests/query_context_tests.py b/tests/integration_tests/query_context_tests.py
@@ -15,6 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 import datetime
+import logging
 import re
 import time
 from typing import Any, Dict
@@ -380,7 +381,7 @@ def test_query_response_type(self):
         assert re.search(r'[`"\[]?num[`"\]]? IS NOT NULL', sql_text)
         assert re.search(
             r"""NOT \([`"\[]?name[`"\]]? IS NULL[\s\n]* """
-            r"""OR [`"\[]?name[`"\]]? IN \('abc'\)\)""",
+            r"""OR [`"\[]?name[`"\]]? IN \('"abc"'\)\)""",
             sql_text,
         )