Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion src/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,8 @@

# pylint: skip-file
# ruff: noqa
if __name__ == "pyhiveapi": # pragma: no cover
# TIM CODE - code generation appears not to fully implement the name change to pyhive - so allow for it
if __name__ in ( "pyhiveapi", "pyhive"): # pragma: no cover
from .api.hive_api import HiveApi as API # type: ignore[assignment] # pragma: no cover
from .api.hive_auth import HiveAuth as Auth # type: ignore[assignment] # pragma: no cover
else:
Expand Down
7 changes: 7 additions & 0 deletions tests/e2e/test_pyhive_exposes_sync_classes.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
from pyhive.api.hive_auth import HiveAuth
from pyhive.api.hive_api import HiveApi
import pyhive

def test_pyhive_exposes_sync_classes():
assert pyhive.Auth is HiveAuth
assert pyhive.API is HiveApi