Skip to content

Commit 5272981

Browse files
committed
fix pylint warnings
1 parent a34001b commit 5272981

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tests/unit/conftest.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,10 @@ class CustomIterator:
6161
def __init__(self, values):
6262
self._values = iter(values)
6363
self._has_next = True
64+
self._next_value = None
6465

65-
def hasNext(self): # pylint: disable=invalid-name
66+
# pylint: disable=invalid-name
67+
def hasNext(self):
6668
try:
6769
self._next_value = next(self._values)
6870
self._has_next = True
@@ -154,9 +156,11 @@ def inner(cb, **replace) -> RuntimeContext:
154156
ctx.tables_crawler._spark._jsparkSession.sharedState().externalCatalog().listDatabases.return_value = (
155157
mock_list_databases_iterator
156158
)
159+
# pylint: disable=protected-access
157160
ctx.tables_crawler._spark._jsparkSession.sharedState().externalCatalog().listTables.return_value = (
158161
mock_list_tables_iterator
159162
)
163+
# pylint: disable=protected-access
160164
ctx.tables_crawler._spark._jsparkSession.sharedState().externalCatalog().getTable.return_value = (
161165
get_table_mock
162166
)
@@ -169,8 +173,9 @@ def inner(cb, **replace) -> RuntimeContext:
169173

170174
@pytest.fixture
171175
def acc_client():
172-
acc = create_autospec(AccountClient) # pylint: disable=mock-no-usage
176+
acc = create_autospec(AccountClient)
173177
acc.config = Config(host="https://accounts.cloud.databricks.com", account_id="123", token="123")
178+
acc.asset_not_called()
174179
return acc
175180

176181

0 commit comments

Comments
 (0)