|
33 | 33 | ): # pragma: no cover |
34 | 34 | DEFAULT_CONSTRUCTORS = default_constructors |
35 | 35 | else: |
36 | | - DEFAULT_CONSTRUCTORS = ( |
37 | | - "pandas,pandas[nullable],pandas[pyarrow],polars[eager],polars[lazy],pyarrow" |
38 | | - ) |
| 36 | + DEFAULT_CONSTRUCTORS = "pandas,pandas[pyarrow],polars[eager],pyarrow,duckdb,sqlframe" |
39 | 37 |
|
40 | 38 |
|
41 | 39 | def pytest_addoption(parser: Any) -> None: |
@@ -232,24 +230,27 @@ def pytest_generate_tests(metafunc: pytest.Metafunc) -> None: |
232 | 230 |
|
233 | 231 | for constructor in selected_constructors: |
234 | 232 | if ( |
235 | | - constructor in {"pandas[nullable]", "pandas[pyarrow]"} |
236 | | - and MIN_PANDAS_NULLABLE_VERSION > PANDAS_VERSION |
237 | | - ) or (constructor == "sqlframe" and sys.version_info < (3, 9)): |
238 | | - continue # pragma: no cover |
239 | | - if constructor == "pyspark" and sys.version_info >= (3, 12): |
| 233 | + ( |
| 234 | + constructor in {"pandas[nullable]", "pandas[pyarrow]"} |
| 235 | + and MIN_PANDAS_NULLABLE_VERSION > PANDAS_VERSION |
| 236 | + ) |
| 237 | + or (constructor == "sqlframe" and sys.version_info < (3, 9)) |
| 238 | + or (constructor == "pyspark" and sys.version_info >= (3, 12)) |
| 239 | + ): |
240 | 240 | continue # pragma: no cover |
241 | 241 |
|
242 | 242 | if constructor in EAGER_CONSTRUCTORS: |
243 | 243 | eager_constructors.append(EAGER_CONSTRUCTORS[constructor]) |
244 | 244 | eager_constructors_ids.append(constructor) |
245 | 245 | constructors.append(EAGER_CONSTRUCTORS[constructor]) |
246 | | - constructors_ids.append(constructor) |
| 246 | + elif constructor == "pyspark": |
| 247 | + constructors.append(pyspark_lazy_constructor()) |
247 | 248 | elif constructor in LAZY_CONSTRUCTORS: |
248 | 249 | constructors.append(LAZY_CONSTRUCTORS[constructor]) |
249 | | - constructors_ids.append(constructor) |
250 | 250 | else: # pragma: no cover |
251 | 251 | msg = f"Expected one of {EAGER_CONSTRUCTORS.keys()} or {LAZY_CONSTRUCTORS.keys()}, got {constructor}" |
252 | 252 | raise ValueError(msg) |
| 253 | + constructors_ids.append(constructor) |
253 | 254 |
|
254 | 255 | if "constructor_eager" in metafunc.fixturenames: |
255 | 256 | metafunc.parametrize( |
|
0 commit comments