feat(rust): access OptState in LazyFrame to unit-test optimizatio…

…n toggle methods. (#9883)
pola-rs · Jul 14, 2023 · 134d43e · 134d43e
1 parent 147944c
commit 134d43e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/polars/polars-lazy/polars-plan/src/frame/opt_state.rs b/polars/polars-lazy/polars-plan/src/frame/opt_state.rs
@@ -1,4 +1,4 @@
-#[derive(Copy, Clone)]
+#[derive(Copy, Clone, Debug)]
 /// State of the allowed optimizations
 pub struct OptState {
     pub projection_pushdown: bool,

diff --git a/polars/polars-lazy/src/frame/mod.rs b/polars/polars-lazy/src/frame/mod.rs
@@ -110,6 +110,11 @@ impl LazyFrame {
         }
     }
 
+    /// Get current optimizations
+    pub fn get_current_optimizations(&self) -> OptState {
+        self.opt_state
+    }
+
     /// Set allowed optimizations
     pub fn with_optimizations(mut self, opt_state: OptState) -> Self {
         self.opt_state = opt_state;