Fix tests

py-polars/tests/unit/io/cloud/test_aws.py

@@ -89,12 +89,12 @@ def test_read_s3(s3: str, function: Callable[..., Any], extension: str) -> None:
     [(pl.scan_ipc, "ipc"), (pl.scan_parquet, "parquet")],
 )
 def test_scan_s3(s3: str, function: Callable[..., Any], extension: str) -> None:
-    df = function(
+    lf = function(
         f"s3://bucket/foods1.{extension}",
         storage_options={"endpoint_url": s3},
     )
-    assert df.columns == ["category", "calories", "fats_g", "sugars_g"]
-    assert df.collect().shape == (27, 4)
+    assert lf.collect_schema().names() == ["category", "calories", "fats_g", "sugars_g"]
+    assert lf.collect().shape == (27, 4)
 
 
 def test_lazy_count_s3(s3: str) -> None:

py-polars/tests/unit/lazyframe/test_lazyframe.py

@@ -14,7 +14,7 @@
 import polars.selectors as cs
 from polars import lit, when
 from polars.datatypes import FLOAT_DTYPES
-from polars.exceptions import PolarsInefficientMapWarning
+from polars.exceptions import PerformanceWarning, PolarsInefficientMapWarning
 from polars.testing import assert_frame_equal, assert_series_equal
 
 if TYPE_CHECKING:
@@ -1382,7 +1382,11 @@ def test_lf_properties() -> None:
             "ham": ["a", "b", "c"],
         }
     )
-    assert lf.schema == {"foo": pl.Int64, "bar": pl.Float64, "ham": pl.String}
-    assert lf.columns == ["foo", "bar", "ham"]
-    assert lf.dtypes == [pl.Int64, pl.Float64, pl.String]
-    assert lf.width == 3
+    with pytest.warns(PerformanceWarning):
+        assert lf.schema == {"foo": pl.Int64, "bar": pl.Float64, "ham": pl.String}
+    with pytest.warns(PerformanceWarning):
+        assert lf.columns == ["foo", "bar", "ham"]
+    with pytest.warns(PerformanceWarning):
+        assert lf.dtypes == [pl.Int64, pl.Float64, pl.String]
+    with pytest.warns(PerformanceWarning):
+        assert lf.width == 3

py-polars/tests/unit/operations/arithmetic/test_arithmetic.py

@@ -670,7 +670,7 @@ def test_arithmetic_duration_div_multiply() -> None:
         e=pl.col("a") * 2.5,
         f=pl.col("a") / pl.col("a"),  # a constant float
     )
-    assert q.schema == pl.Schema(
+    assert q.collect_schema() == pl.Schema(
         [
             ("a", pl.Duration(time_unit="us")),
             ("b", pl.Duration(time_unit="us")),
@@ -715,7 +715,7 @@ def test_arithmetic_duration_div_multiply() -> None:
         b=2 * pl.col("a"),
         c=2.5 * pl.col("a"),
     )
-    assert q.schema == pl.Schema(
+    assert q.collect_schema() == pl.Schema(
         [
             ("a", pl.Duration(time_unit="us")),
             ("b", pl.Duration(time_unit="us")),