Skip to content

Commit

Permalink
actually let's fail it
Browse files Browse the repository at this point in the history
  • Loading branch information
douglasdavis committed Sep 6, 2023
1 parent 13933ae commit 75d65fd
Showing 1 changed file with 2 additions and 13 deletions.
15 changes: 2 additions & 13 deletions tests/test_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -830,22 +830,11 @@ def test_map_partitions_no_dask_collections_passed(caa):


@pytest.mark.parametrize("fn", [dak.count, dak.zeros_like, dak.ones_like])
@pytest.mark.parametrize(
"opt_enabled",
[
pytest.param(
True, marks=pytest.mark.xfail(reason="these fail with opt enabled")
),
False,
],
)
def test_shape_only_ops(
fn: Callable, opt_enabled: bool, tmp_path_factory: pytest.TempPathFactory
) -> None:
def test_shape_only_ops(fn: Callable, tmp_path_factory: pytest.TempPathFactory) -> None:
a = ak.Array([{"a": 1, "b": 2}, {"a": 3, "b": 4}])
p = tmp_path_factory.mktemp("zeros-like-flat")
ak.to_parquet(a, str(p / "file.parquet"))
lazy = dak.from_parquet(str(p))
result = fn(lazy.b)
with dask.config.set({"awkward.optimization.enabled": opt_enabled}):
with dask.config.set({"awkward.optimization.enabled": True}):
result.compute()

0 comments on commit 75d65fd

Please sign in to comment.