Generalized Raw* class checking

Author: Spaarsh

python/tests/test_wrapper_coverage.py

@@ -34,14 +34,13 @@ def missing_exports(internal_obj, wrapped_obj) -> None:
         return
 
     for attr in dir(internal_obj):
-        # Skip internal context and RawExpr (which is handled by Expr class)
-        if attr in ["_global_ctx", "RawExpr"]:
+        if attr in ["_global_ctx"]:
             continue
 
-        # Check if RawExpr functionality is covered by Expr class
-        if attr == "RawExpr" and hasattr(wrapped_obj, "Expr"):
-            expr_class = getattr(wrapped_obj, "Expr")
-            assert hasattr(expr_class, "raw_expr"), "Expr class must provide raw_expr property"
+        # 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)