add __version__ attribute to datafusion object

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",

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,