diff --git a/datafusion/__init__.py b/datafusion/__init__.py index 0a25592..8174380 100644 --- a/datafusion/__init__.py +++ b/datafusion/__init__.py @@ -18,6 +18,7 @@ from abc import ABCMeta, abstractmethod from typing import List +import importlib.metadata import pyarrow as pa from ._internal import ( @@ -29,6 +30,9 @@ ) +__version__ = importlib.metadata.version("datafusion") + + __all__ = [ "DataFrame", "ExecutionContext", diff --git a/datafusion/tests/test_imports.py b/datafusion/tests/test_imports.py index 4238002..2a3a21c 100644 --- a/datafusion/tests/test_imports.py +++ b/datafusion/tests/test_imports.py @@ -32,6 +32,10 @@ def test_import_datafusion(): assert datafusion.__name__ == "datafusion" +def test_datafusion_python_version(): + assert datafusion.__version__ is not None + + def test_class_module_is_datafusion(): for klass in [ ExecutionContext,