summaryrefslogtreecommitdiffstats
path: root/tests/test_executor.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2023-10-16 11:37:35 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2023-10-16 11:37:35 +0000
commit45eab59f38aae326a4970767f9bc7119bef339b9 (patch)
treef5a21c1bba9fd891c923cc14206986691931ca47 /tests/test_executor.py
parentAdding upstream version 18.11.6. (diff)
downloadsqlglot-45eab59f38aae326a4970767f9bc7119bef339b9.tar.xz
sqlglot-45eab59f38aae326a4970767f9bc7119bef339b9.zip
Adding upstream version 18.13.0.upstream/18.13.0
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'tests/test_executor.py')
-rw-r--r--tests/test_executor.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/tests/test_executor.py b/tests/test_executor.py
index c6b85c9..721550e 100644
--- a/tests/test_executor.py
+++ b/tests/test_executor.py
@@ -4,6 +4,7 @@ from datetime import date
from multiprocessing import Pool
import duckdb
+import numpy as np
import pandas as pd
from pandas.testing import assert_frame_equal
@@ -94,6 +95,11 @@ class TestExecutor(unittest.TestCase):
sql, _ = self.sqls[i]
a = self.cached_execute(sql)
b = pd.DataFrame(table.rows, columns=table.columns)
+
+ # The executor represents NULL values as None, whereas DuckDB represents them as NaN,
+ # and so the following is done to silence Pandas' "Mismatched null-like values" warnings
+ b = b.fillna(value=np.nan)
+
assert_frame_equal(a, b, check_dtype=False, check_index_type=False)
def test_execute_callable(self):