Move marker to streaming module level

pola-rs · Jan 24, 2024 · f6f46c0 · f6f46c0
1 parent 907ceb0
commit f6f46c0
Show file tree

Hide file tree

Showing 9 changed files with 5 additions and 22 deletions.
diff --git a/py-polars/tests/unit/streaming/__init__.py b/py-polars/tests/unit/streaming/__init__.py
@@ -0,0 +1,3 @@
+import pytest
+
+pytestmark = pytest.mark.xdist_group("streaming")
diff --git a/py-polars/tests/unit/streaming/test_streaming.py b/py-polars/tests/unit/streaming/test_streaming.py
@@ -17,8 +17,6 @@
 if TYPE_CHECKING:
     from polars.type_aliases import JoinStrategy
 
-pytestmark = pytest.mark.xdist_group("streaming")
-
 
 def test_streaming_categoricals_5921() -> None:
     with pl.StringCache():

diff --git a/py-polars/tests/unit/streaming/test_streaming_categoricals.py b/py-polars/tests/unit/streaming/test_streaming_categoricals.py
@@ -1,9 +1,5 @@
-import pytest
-
 import polars as pl
 
-pytestmark = pytest.mark.xdist_group("streaming")
-
 
 def test_streaming_nested_categorical() -> None:
     assert (

diff --git a/py-polars/tests/unit/streaming/test_streaming_cse.py b/py-polars/tests/unit/streaming/test_streaming_cse.py
@@ -7,8 +7,6 @@
 import polars as pl
 from polars.testing import assert_frame_equal
 
-pytestmark = pytest.mark.xdist_group("streaming")
-
 
 def test_cse_expr_selection_streaming(monkeypatch: Any, capfd: Any) -> None:
     monkeypatch.setenv("POLARS_VERBOSE", "1")

diff --git a/py-polars/tests/unit/streaming/test_streaming_group_by.py b/py-polars/tests/unit/streaming/test_streaming_group_by.py
@@ -9,8 +9,6 @@
 import polars as pl
 from polars.testing import assert_frame_equal
 
-pytestmark = pytest.mark.xdist_group("streaming")
-
 
 @pytest.mark.slow()
 def test_streaming_group_by_sorted_fast_path_nulls_10273() -> None:

diff --git a/py-polars/tests/unit/streaming/test_streaming_io.py b/py-polars/tests/unit/streaming/test_streaming_io.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
-import unittest
 from typing import TYPE_CHECKING
+from unittest.mock import patch
 
 import pytest
 
@@ -12,9 +12,6 @@
     from pathlib import Path
 
 
-pytestmark = pytest.mark.xdist_group("streaming")
-
-
 @pytest.mark.write_disk()
 def test_streaming_parquet_glob_5900(df: pl.DataFrame, tmp_path: Path) -> None:
     tmp_path.mkdir(exist_ok=True)
@@ -122,7 +119,7 @@ def test_sink_csv_with_options() -> None:
      passed into the rust-polars correctly.
     """
     df = pl.LazyFrame({"dummy": ["abc"]})
-    with unittest.mock.patch.object(df, "_ldf") as ldf:
+    with patch.object(df, "_ldf") as ldf:
         df.sink_csv(
             "path",
             include_bom=True,

diff --git a/py-polars/tests/unit/streaming/test_streaming_join.py b/py-polars/tests/unit/streaming/test_streaming_join.py
@@ -4,13 +4,10 @@
 
 import numpy as np
 import pandas as pd
-import pytest
 
 import polars as pl
 from polars.testing import assert_frame_equal
 
-pytestmark = pytest.mark.xdist_group("streaming")
-
 
 def test_streaming_joins() -> None:
     n = 100

diff --git a/py-polars/tests/unit/streaming/test_streaming_sort.py b/py-polars/tests/unit/streaming/test_streaming_sort.py
@@ -13,8 +13,6 @@
 if TYPE_CHECKING:
     from pathlib import Path
 
-pytestmark = pytest.mark.xdist_group("streaming")
-
 
 def assert_df_sorted_by(
     df: pl.DataFrame,

diff --git a/py-polars/tests/unit/streaming/test_streaming_unique.py b/py-polars/tests/unit/streaming/test_streaming_unique.py
@@ -10,8 +10,6 @@
 if TYPE_CHECKING:
     from pathlib import Path
 
-pytestmark = pytest.mark.xdist_group("streaming")
-
 
 @pytest.mark.write_disk()
 @pytest.mark.slow()