BUG: during interchanging from non-pandas tz-aware data (#54287)

* TST: add a test for interchanging from non-pandas tz-aware

* TST: correct the test for interchanging from non-pandas tz-aware

* check an error in the test

* correct test

* add extra except in set_nulls

* fix bug during interchanging from non-pandas tz-aware

* correct test

* correct wording

Author: natmokval

doc/source/whatsnew/v2.1.0.rst

@@ -779,6 +779,7 @@ Other
 - Bug in :class:`DatetimeIndex` where ``repr`` of index passed with time does not print time is midnight and non-day based freq(:issue:`53470`)
 - Bug in :class:`FloatingArray.__contains__` with ``NaN`` item incorrectly returning ``False`` when ``NaN`` values are present (:issue:`52840`)
 - Bug in :func:`api.interchange.from_dataframe` was not respecting ``allow_copy`` argument (:issue:`54322`)
+- Bug in :func:`api.interchange.from_dataframe` was raising during interchanging from non-pandas tz-aware data containing null values (:issue:`54287`)
 - Bug in :func:`api.interchange.from_dataframe` when converting an empty DataFrame object (:issue:`53155`)
 - Bug in :func:`assert_almost_equal` now throwing assertion error for two unequal sets (:issue:`51727`)
 - Bug in :func:`assert_frame_equal` checks category dtypes even when asked not to check index type (:issue:`52126`)

pandas/core/interchange/from_dataframe.py

@@ -7,6 +7,7 @@
 import numpy as np
 
 from pandas.compat._optional import import_optional_dependency
+from pandas.errors import SettingWithCopyError
 
 import pandas as pd
 from pandas.core.interchange.dataframe_protocol import (
@@ -514,5 +515,9 @@ def set_nulls(
             # cast the `data` to nullable float dtype.
             data = data.astype(float)
             data[null_pos] = None
+        except SettingWithCopyError:
+            # `SettingWithCopyError` may happen for datetime-like with missing values.
+            data = data.copy()
+            data[null_pos] = None
 
     return data

pandas/tests/interchange/test_impl.py

@@ -308,3 +308,22 @@ def test_datetimetzdtype(tz, unit):
     )
     df = pd.DataFrame({"ts_tz": tz_data})
     tm.assert_frame_equal(df, from_dataframe(df.__dataframe__()))
+
+
+def test_interchange_from_non_pandas_tz_aware():
+    # GH 54239, 54287
+    pa = pytest.importorskip("pyarrow", "11.0.0")
+    import pyarrow.compute as pc
+
+    arr = pa.array([datetime(2020, 1, 1), None, datetime(2020, 1, 2)])
+    arr = pc.assume_timezone(arr, "Asia/Kathmandu")
+    table = pa.table({"arr": arr})
+    exchange_df = table.__dataframe__()
+    result = from_dataframe(exchange_df)
+
+    expected = pd.DataFrame(
+        ["2020-01-01 00:00:00+05:45", "NaT", "2020-01-02 00:00:00+05:45"],
+        columns=["arr"],
+        dtype="datetime64[us, Asia/Kathmandu]",
+    )
+    tm.assert_frame_equal(expected, result)