forked from pool/python-sqlglot
51 lines
1.2 KiB
Diff
51 lines
1.2 KiB
Diff
|
---
|
||
|
tests/test_executor.py | 7 ++++++-
|
||
|
tests/test_optimizer.py | 7 ++++++-
|
||
|
2 files changed, 12 insertions(+), 2 deletions(-)
|
||
|
|
||
|
--- a/tests/test_executor.py
|
||
|
+++ b/tests/test_executor.py
|
||
|
@@ -1,6 +1,10 @@
|
||
|
import unittest
|
||
|
|
||
|
-import duckdb
|
||
|
+try:
|
||
|
+ import duckdb
|
||
|
+ DUCKDB_AVAILABLE = True
|
||
|
+except ImportError:
|
||
|
+ DUCKDB_AVAILABLE = False
|
||
|
import pandas as pd
|
||
|
from pandas.testing import assert_frame_equal
|
||
|
|
||
|
@@ -12,6 +16,7 @@ from tests.helpers import FIXTURES_DIR,
|
||
|
DIR = FIXTURES_DIR + "/optimizer/tpc-h/"
|
||
|
|
||
|
|
||
|
+@unittest.skipUnless(DUCKDB_AVAILABLE, "Cannot work without duckdb.")
|
||
|
class TestExecutor(unittest.TestCase):
|
||
|
@classmethod
|
||
|
def setUpClass(cls):
|
||
|
--- a/tests/test_optimizer.py
|
||
|
+++ b/tests/test_optimizer.py
|
||
|
@@ -1,7 +1,11 @@
|
||
|
import unittest
|
||
|
from functools import partial
|
||
|
|
||
|
-import duckdb
|
||
|
+try:
|
||
|
+ import duckdb
|
||
|
+ DUCKDB_AVAILABLE = True
|
||
|
+except ImportError:
|
||
|
+ DUCKDB_AVAILABLE = False
|
||
|
from pandas.testing import assert_frame_equal
|
||
|
|
||
|
import sqlglot
|
||
|
@@ -18,6 +22,7 @@ from tests.helpers import (
|
||
|
)
|
||
|
|
||
|
|
||
|
+@unittest.skipUnless(DUCKDB_AVAILABLE, "Cannot work without duckdb.")
|
||
|
class TestOptimizer(unittest.TestCase):
|
||
|
maxDiff = None
|
||
|
|