Renaming Internal Structs

python/datafusion/expr.py

@@ -102,6 +102,7 @@
 
 __all__ = [
     "Expr",
+    "RawExpr",
     "Column",
     "Literal",
     "BinaryExpr",
@@ -193,10 +194,15 @@
     :ref:`Expressions` in the online documentation for more information.
     """
 
-    def __init__(self, expr: expr_internal.Expr) -> None:
+    def __init__(self, expr: expr_internal.RawExpr) -> None:
         """This constructor should not be called by the end user."""
         self.expr = expr
 
+    @property 
+    def raw_expr(self) -> expr_internal.RawExpr:
+        """Returns the underlying RawExpr instance."""
+        return self.expr
+
     def to_variant(self) -> Any:
         """Convert this expression into a python object if possible."""
         return self.expr.to_variant()
@@ -383,7 +389,7 @@
             value = pa.scalar(value, type=pa.string_view())
         if not isinstance(value, pa.Scalar):
             value = pa.scalar(value)
-        return Expr(expr_internal.Expr.literal(value))
+        return Expr(expr_internal.RawExpr.literal(value))
 
     @staticmethod
     def string_literal(value: str) -> Expr:
@@ -398,13 +404,13 @@
         """
         if isinstance(value, str):
             value = pa.scalar(value, type=pa.string())
-            return Expr(expr_internal.Expr.literal(value))
+            return Expr(expr_internal.RawExpr.literal(value))
         return Expr.literal(value)
 
     @staticmethod
     def column(value: str) -> Expr:
         """Creates a new expression representing a column."""
-        return Expr(expr_internal.Expr.column(value))
+        return Expr(expr_internal.RawExpr.column(value))
 
     def alias(self, name: str) -> Expr:
         """Assign a name to the expression."""

python/tests/test_wrapper_coverage.py

@@ -36,7 +36,12 @@
     for attr in dir(internal_obj):
         if attr in ["_global_ctx"]:
             continue
-        assert attr in dir(wrapped_obj)
+
+        # Check if Raw* classes have corresponding wrapper classes
+        if attr.startswith("Raw"):
+            base_class = attr[3:]  # Remove "Raw" prefix
+            assert hasattr(wrapped_obj, base_class), f"Missing implementation for {attr}"
+            continue
 
         internal_attr = getattr(internal_obj, attr)
         wrapped_attr = getattr(wrapped_obj, attr)

src/expr.rs

@@ -101,7 +101,7 @@ pub mod window;
 use sort_expr::{to_sort_expressions, PySortExpr};
 
 /// A PyExpr that can be used on a DataFrame
-#[pyclass(name = "Expr", module = "datafusion.expr", subclass)]
+#[pyclass(name = "RawExpr", module = "datafusion.expr", subclass)]
 #[derive(Debug, Clone)]
 pub struct PyExpr {
     pub expr: Expr,