From 420085e6ac2cd5d03c7b9923f629de4566a6a7bc Mon Sep 17 00:00:00 2001
From: Jack Wright <jack.wright@disqo.com>
Date: Mon, 8 Jul 2024 11:46:21 -0700
Subject: [PATCH] Added an example and test to `polars pivot`

---
 .../src/dataframe/eager/pivot.rs              | 30 +++++++++++++++++--
 1 file changed, 28 insertions(+), 2 deletions(-)

diff --git a/crates/nu_plugin_polars/src/dataframe/eager/pivot.rs b/crates/nu_plugin_polars/src/dataframe/eager/pivot.rs
index 360745f183..b72210d011 100644
--- a/crates/nu_plugin_polars/src/dataframe/eager/pivot.rs
+++ b/crates/nu_plugin_polars/src/dataframe/eager/pivot.rs
@@ -8,7 +8,7 @@ use polars_ops::pivot::{pivot, PivotAgg};
 
 use crate::{
     dataframe::values::utils::convert_columns_string,
-    values::{CustomValueSupport, PolarsPluginObject},
+    values::{Column, CustomValueSupport, PolarsPluginObject},
     PolarsPlugin,
 };
 
@@ -72,7 +72,33 @@ impl PluginCommand for PivotDF {
     }
 
     fn examples(&self) -> Vec<Example> {
-        vec![]
+        vec![
+            Example {
+                example: "[[name subject test_1 test_2]; [Cady maths 98 100] [Cady physics 99 100] [Karen maths 61 60] [Karen physics 58 60]] | polars into-df |  polars pivot --on [subject] --index [name] --values [test_1]",
+                description: "Perform a pivot in order to show individuals test score by subject", 
+                result: Some(
+                    NuDataFrame::try_from_columns(
+                        vec![
+                            Column::new(
+                                "name".to_string(),
+                                vec![Value::string("Cady", Span::test_data()), Value::string("Karen", Span::test_data())],
+                            ),
+                            Column::new(
+                                "maths".to_string(),
+                                vec![Value::int(98, Span::test_data()), Value::int(61, Span::test_data())],
+                            ),
+                            Column::new(
+                                "physics".to_string(),
+                                vec![Value::int(99, Span::test_data()), Value::int(58, Span::test_data())],
+                            ),
+                        ],
+                        None,
+                    )
+                    .expect("simple df for test should not fail")
+                    .into_value(Span::unknown())
+                )
+            }
+        ]
     }
 
     fn run(