code: def test_duckdb_timestamp_with_timezone():
      try:
          vf.runtime.set_connection("duckdb")
          dates_df = pd.DataFrame(
              {
                  "date_col": [
                      date(2022, 1, 1),
                      date(2022, 1, 2),
                      date(2022, 1, 3),
                  ],
              }
          )
          dates_df["date_col"] = pd.to_datetime(
              dates_df.date_col
          ).dt.tz_localize("UTC")
          spec = date_column_spec()
  
          (output_ds,), warnings = vf.runtime.pre_transform_datasets(
              spec,
              ["data_0"],
              "America/New_York",
              default_input_tz="UTC",
              inline_datasets=dict(dates=dates_df),
          )
  
          # Timestamps are in the local timezone, so they should be midnight local time
          assert list(output_ds.date_col) == [
              pd.Timestamp("2022-01-01 00:00:00", tz="UTC"),
              pd.Timestamp("2022-01-02 00:00:00", tz="UTC"),
              pd.Timestamp("2022-01-03 00:00:00", tz="UTC"),
          ]
      finally:
          vf.runtime.set_connection("datafusion")
  
imports: import pandas
  import pytest
  from pandas import Timestamp, NaT
  import pyarrow
  import vegafusion
  import json
  import polars
  from datetime import date
  import decimal
  import base64
label_desc: Use image node_id 25 for calling a global function with example usage: test_duckdb_timestamp_with_timezone() without return types