test: cover new DF54 expr wrappers, catalog factories, and DataFrame.alias

Add module-metadata checks for HigherOrderFunction, Lambda, LambdaVariable
and the top-level TableProviderFactory / TableProviderFactoryExportable
re-exports, plus a self-join regression test exercising the new
DataFrame.alias() qualifier-based selection path.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Author: timsaucer

python/tests/test_dataframe.py

@@ -292,6 +292,20 @@ def test_select_exprs(df):
     assert result.column(1) == pa.array([3, 3, 3])
 
 
+def test_alias_self_join(df):
+    left = df.alias("l")
+    right = df.alias("r")
+    joined = left.join(right, left_on="a", right_on="a").select(
+        "a",
+        column("l.b").alias("lb"),
+        column("r.b").alias("rb"),
+    )
+    result = joined.sort(column("a")).collect()[0]
+    assert result.column(0) == pa.array([1, 2, 3])
+    assert result.column(1) == pa.array([4, 5, 6])
+    assert result.column(2) == pa.array([4, 5, 6])
+
+
 def test_drop_quoted_columns():
     ctx = SessionContext()
     batch = pa.RecordBatch.from_arrays([pa.array([1, 2, 3])], names=["ID_For_Students"])

python/tests/test_imports.py

@@ -22,6 +22,8 @@
     DataFrame,
     ScalarUDF,
     SessionContext,
+    TableProviderFactory,
+    TableProviderFactoryExportable,
     functions,
 )
 from datafusion.common import (
@@ -47,6 +49,7 @@
     Extension,
     Filter,
     GroupingSet,
+    HigherOrderFunction,
     ILike,
     InList,
     InSubquery,
@@ -60,6 +63,8 @@
     Join,
     JoinConstraint,
     JoinType,
+    Lambda,
+    LambdaVariable,
     Like,
     Limit,
     Literal,
@@ -160,9 +165,16 @@ def test_class_module_is_datafusion():
         DropTable,
         Repartition,
         Partitioning,
+        HigherOrderFunction,
+        Lambda,
+        LambdaVariable,
     ]:
         assert klass.__module__ == "datafusion.expr"
 
+    # catalog factory types (re-exported at top level)
+    for klass in [TableProviderFactory, TableProviderFactoryExportable]:
+        assert klass.__module__ == "datafusion.catalog"
+
     # schema
     for klass in [DFSchema]:
         assert klass.__module__ == "datafusion.common"