test: cover v1, v2

Author: dangotbanned

tests/v1_test.py

@@ -1,6 +1,7 @@
 # Test assorted functions which we overwrite in stable.v1
 from __future__ import annotations
 
+from collections import deque
 from contextlib import nullcontext as does_not_raise
 from datetime import datetime, timedelta
 from typing import TYPE_CHECKING, Any, Callable, cast
@@ -37,6 +38,7 @@
     Constructor,
     ConstructorEager,
     assert_equal_data,
+    assert_equal_series,
 )
 
 if TYPE_CHECKING:
@@ -1049,3 +1051,30 @@ def test_series_from_numpy(
     if dtype:
         assert result.dtype == dtype
     assert_equal_data(result.to_frame(), {name: expected})
+
+
+@pytest.mark.parametrize(
+    ("dtype", "expected"),
+    [
+        (None, [5, 2, 0, 1]),
+        (nw_v1.Int64, [5, 2, 0, 1]),
+        (nw_v1.String, ("a", "b", "c")),
+        (nw_v1.Float64, [5.0, 2.0, 0.0, 1.0]),
+        (
+            nw_v1.Datetime(),
+            deque([datetime(2005, 1, 1, 10), datetime(2002, 1, 1, 10, 43)]),
+        ),
+    ],
+    ids=str,
+)
+def test_series_from_iterable(
+    eager_backend: EagerAllowed, dtype: IntoDType | None, expected: Sequence[Any]
+) -> None:
+    data = expected
+    name = "abc"
+    result = nw_v1.Series.from_iterable(name, data, backend=eager_backend, dtype=dtype)
+    assert result._version is Version.V1
+    assert isinstance(result, nw_v1.Series)
+    if dtype:
+        assert result.dtype == dtype
+    assert_equal_series(result, expected, name)

tests/v2_test.py

@@ -2,6 +2,8 @@
 
 from __future__ import annotations
 
+from collections import deque
+from datetime import datetime
 from typing import TYPE_CHECKING, Any
 
 import numpy as np
@@ -10,12 +12,21 @@
 
 import narwhals.stable.v2 as nw_v2
 from narwhals.utils import Version
-from tests.utils import PANDAS_VERSION, Constructor, assert_equal_data
+from tests.utils import (
+    PANDAS_VERSION,
+    Constructor,
+    assert_equal_data,
+    assert_equal_series,
+)
 
 if TYPE_CHECKING:
+    from collections.abc import Sequence
+
     from typing_extensions import assert_type
 
+    from narwhals._namespace import EagerAllowed
     from narwhals.stable.v2.typing import IntoDataFrameT
+    from narwhals.typing import IntoDType
 
 
 def test_toplevel() -> None:
@@ -332,3 +343,30 @@ def test_imports() -> None:
     from narwhals.stable.v2.dtypes import Enum  # noqa: F401
     from narwhals.stable.v2.selectors import datetime  # noqa: F401
     from narwhals.stable.v2.typing import IntoDataFrame  # noqa: F401
+
+
+@pytest.mark.parametrize(
+    ("dtype", "expected"),
+    [
+        (None, [5, 2, 0, 1]),
+        (nw_v2.Int64, [5, 2, 0, 1]),
+        (nw_v2.String, ("a", "b", "c")),
+        (nw_v2.Float64, [5.0, 2.0, 0.0, 1.0]),
+        (
+            nw_v2.Datetime(),
+            deque([datetime(2005, 1, 1, 10), datetime(2002, 1, 1, 10, 43)]),
+        ),
+    ],
+    ids=str,
+)
+def test_series_from_iterable(
+    eager_backend: EagerAllowed, dtype: IntoDType | None, expected: Sequence[Any]
+) -> None:
+    data = expected
+    name = "abc"
+    result = nw_v2.Series.from_iterable(name, data, backend=eager_backend, dtype=dtype)
+    assert result._version is Version.V2
+    assert isinstance(result, nw_v2.Series)
+    if dtype:
+        assert result.dtype == dtype
+    assert_equal_series(result, expected, name)