diff options
Diffstat (limited to 'tests/dataframe')
| -rw-r--r-- | tests/dataframe/unit/test_functions.py | 3 | ||||
| -rw-r--r-- | tests/dataframe/unit/test_session.py | 3 |
2 files changed, 2 insertions, 4 deletions
diff --git a/tests/dataframe/unit/test_functions.py b/tests/dataframe/unit/test_functions.py index f155065..d9a32c4 100644 --- a/tests/dataframe/unit/test_functions.py +++ b/tests/dataframe/unit/test_functions.py @@ -2,8 +2,7 @@ import datetime import inspect import unittest -from sqlglot import expressions as exp -from sqlglot import parse_one +from sqlglot import expressions as exp, parse_one from sqlglot.dataframe.sql import functions as SF from sqlglot.errors import ErrorLevel diff --git a/tests/dataframe/unit/test_session.py b/tests/dataframe/unit/test_session.py index f5b79fd..7da0833 100644 --- a/tests/dataframe/unit/test_session.py +++ b/tests/dataframe/unit/test_session.py @@ -1,8 +1,7 @@ from unittest import mock import sqlglot -from sqlglot.dataframe.sql import functions as F -from sqlglot.dataframe.sql import types +from sqlglot.dataframe.sql import functions as F, types from sqlglot.dataframe.sql.session import SparkSession from sqlglot.schema import MappingSchema from tests.dataframe.unit.dataframe_sql_validator import DataFrameSQLValidator |