Skip to content

Commit 39bb6b3

Browse files
committed
test(typing): Add pa.ChunkedArray into the mix
1 parent d5a126d commit 39bb6b3

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

tests/translate/from_native_test.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -453,14 +453,17 @@ def test_eager_only_pass_through_main(constructor: Constructor) -> None:
453453
nw.from_native(df, eager_only=True, pass_through=False) # type: ignore[type-var]
454454

455455

456-
def test_from_native_eager_only_series_only_allow() -> None:
456+
def test_from_native_eager_only_series_only_allow() -> None: # noqa: PLR0914
457457
pytest.importorskip("polars")
458458
pytest.importorskip("pandas")
459+
pytest.importorskip("pyarrow")
459460
import pandas as pd
460461
import polars as pl
462+
import pyarrow as pa
461463

462464
pl_ser = pl.Series([1, 2, 3])
463465
pd_ser = pd.Series([1, 2, 3])
466+
pa_ser = pa.chunked_array([pa.array([1])])
464467

465468
s01 = nw.from_native(pl_ser, series_only=True)
466469
s02 = nw.from_native(pl_ser, allow_series=True)
@@ -489,3 +492,17 @@ def test_from_native_eager_only_series_only_allow() -> None:
489492
assert isinstance(s14, nw.Series)
490493
assert isinstance(s15, nw.Series)
491494
assert isinstance(s16, nw.Series)
495+
496+
s21 = nw.from_native(pa_ser, series_only=True)
497+
s22 = nw.from_native(pa_ser, allow_series=True)
498+
s23 = nw.from_native(pa_ser, eager_only=True, series_only=True)
499+
s24 = nw.from_native(pa_ser, eager_only=True, series_only=True, allow_series=True)
500+
s25 = nw.from_native(pa_ser, eager_only=True, allow_series=True)
501+
s26 = nw.from_native(pa_ser, series_only=True, allow_series=True)
502+
503+
assert isinstance(s21, nw.Series)
504+
assert isinstance(s22, nw.Series)
505+
assert isinstance(s23, nw.Series)
506+
assert isinstance(s24, nw.Series)
507+
assert isinstance(s25, nw.Series)
508+
assert isinstance(s26, nw.Series)

0 commit comments

Comments
 (0)