c

pola-rs · Oct 22, 2024 · f9ff9eb · f9ff9eb
1 parent e8edca2
commit f9ff9eb
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/py-polars/tests/unit/io/cloud/test_cloud.py b/py-polars/tests/unit/io/cloud/test_cloud.py
@@ -91,13 +91,12 @@ def test_scan_credential_provider_serialization_pyversion() -> None:
 
     v = b"PLPYFN"
     i = serialized.index(v) + len(v)
-    a, b, *_ = serialized[i:]
+    a, b = serialized[i:][:2]
     serialized_pyver = (a, b)
     assert serialized_pyver == (sys.version_info.minor, sys.version_info.micro)
+    # Note: These are loaded as u8's
     serialized[i] = 255
-    serialized[i + 1] = 255
+    serialized[i + 1] = 254
 
-    with pytest.raises(
-        ComputeError, match="python version .* (3, 255, 255) .* differs .*"
-    ):
+    with pytest.raises(ComputeError, match=r"python version.*(3, 255, 254).*differs.*"):
         lf = pl.LazyFrame.deserialize(io.BytesIO(serialized))