From 2fb8c07ec4ce6b8817e0e1a7a6c4cb09fd75a237 Mon Sep 17 00:00:00 2001
From: Spaarsh-root <spaarshthakkar11010@gmail.com>
Date: Mon, 17 Mar 2025 20:33:36 +0530
Subject: [PATCH] Add ctx = SessionContext() to

---
 python/datafusion/__init__.py |  3 +++
 python/tests/test_context.py  | 13 +++++++++++++
 2 files changed, 16 insertions(+)

diff --git a/python/datafusion/__init__.py b/python/datafusion/__init__.py
index 286e5dc31..9e5d08ea8 100644
--- a/python/datafusion/__init__.py
+++ b/python/datafusion/__init__.py
@@ -50,6 +50,8 @@
 from .record_batch import RecordBatch, RecordBatchStream
 from .udf import Accumulator, AggregateUDF, ScalarUDF, WindowUDF, udaf, udf, udwf
 
+ctx = SessionContext()
+
 __version__ = importlib_metadata.version(__name__)
 
 __all__ = [
@@ -76,6 +78,7 @@
     "col",
     "column",
     "common",
+    "ctx",
     "expr",
     "functions",
     "lit",
diff --git a/python/tests/test_context.py b/python/tests/test_context.py
index 4a15ac9cf..fc04e88ac 100644
--- a/python/tests/test_context.py
+++ b/python/tests/test_context.py
@@ -36,6 +36,19 @@ def test_create_context_no_args():
     SessionContext()
 
 
+def test_default_ctx_instance():
+    """Test that the default ctx instance works correctly"""
+    from datafusion import SessionContext, ctx
+
+    # Test that ctx is an instance of SessionContext
+    assert isinstance(ctx, SessionContext)
+
+    # Test basic functionality
+    df = ctx.sql("SELECT 1 as num")
+    result = df.collect()
+    assert len(result) == 1
+
+
 def test_create_context_session_config_only():
     SessionContext(config=SessionConfig())