add __version__ attribute to datafusion object (#3)

datafusion-contrib · Jan 5, 2022 · da792c2 · da792c2
1 parent 5ad043e
commit da792c2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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,