Capture AST for DataFrame.pivot and DataFrame.unpivot

[sfc-gh-oplaton]

1. Which Jira issue is this PR addressing? Make sure that there is an accompanying issue to your PR.

   Fixes SNOW-1491297

2. Fill out the following pre-review checklist:

   • I am adding a new automated test(s) to verify correctness of my new code
     • If this test skips Local Testing mode, I'm requesting review from @snowflakedb/local-testing
   • I am adding new logging messages
   • I am adding a new telemetry message
   • I am adding new credentials
   • I am adding a new dependency
   • If this is a new feature/behavior, I'm adding the Local Testing parity changes.

3. Please describe how your code solves the related issue.

Capture AST for DataFrame.pivot and DataFrame.unpivot.
Add expectation test for pivot.
I was unable to resolve the local testing issue for unpivot. Adding a disabled test for now, and will revisit the test issue separately.

[sfc-gh-lspiegelberg]

Few nits, do you have by chance the error that happens in local testing mode for unpivot?

[sfc-gh-oplaton]

Few nits, do you have by chance the error that happens in local testing mode for unpivot?

The verbatim error is:

session = <snowflake.snowpark.session.Session object at 0x16ddc9e80>
test_case = TestCase(filename='DataFrame.unpivot.test', source='\ndf = session.create_dataframe(\n    [(1, "electronics", 100, 200... df.unpivot("sales", "month", ["jan", "feb"]).sort("empid")\n\n\n', expected_ast_base64='', expected_ast_unparsed='\n')

    @pytest.mark.parametrize("test_case", load_test_cases(), ids=idfn)
    def test_ast(session, test_case):
>       actual, base64_str = run_test(
            session, test_case.filename.replace(".", "_"), test_case.source
        )

/Users/oplaton/src/snowparkpy/tests/ast/test_ast_driver.py:252: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/Users/oplaton/src/snowparkpy/tests/ast/test_ast_driver.py:243: in run_test
    base64_batches = test_module.run_test(session)
/var/folders/5r/s68_pbc972lfk7s9qms41w1c0000gn/T/test_asthorw47_6.py:94: in run_test
    ???
/Users/oplaton/src/snowparkpy/src/snowflake/snowpark/_internal/telemetry.py:212: in wrap
    r = func(*args, **kwargs)
/Users/oplaton/src/snowparkpy/src/snowflake/snowpark/dataframe.py:2263: in unpivot
    from_=SelectSnowflakePlan(
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <snowflake.snowpark._internal.analyzer.select_statement.SelectSnowflakePlan object at 0x16dd3a640>
snowflake_plan = <snowflake.snowpark._internal.analyzer.unary_plan_node.Unpivot object at 0x16e05dac0>

    def __init__(self, snowflake_plan: LogicalPlan, *, analyzer: "Analyzer") -> None:
        super().__init__(analyzer)
        self._snowflake_plan: SnowflakePlan = (
            snowflake_plan
            if isinstance(snowflake_plan, SnowflakePlan)
            else analyzer.resolve(snowflake_plan)
        )
>       self.expr_to_alias.update(self._snowflake_plan.expr_to_alias)
E       AttributeError: 'NoneType' object has no attribute 'expr_to_alias'

/Users/oplaton/src/snowparkpy/src/snowflake/snowpark/_internal/analyzer/select_statement.py:552: AttributeError

[sfc-gh-lspiegelberg]

Great work!