Skip to content

Commit

Permalink
Remove wrappers around ds.rename and ds.drop_vars in conftest.py
Browse files Browse the repository at this point in the history
  • Loading branch information
lochhh committed Aug 7, 2024
1 parent cdf0550 commit 4bfb237
Showing 1 changed file with 6 additions and 24 deletions.
30 changes: 6 additions & 24 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -390,16 +390,6 @@ def valid_poses_dataset_with_nan(valid_poses_dataset):
return valid_poses_dataset


def rename_dimension_in_ds(valid_dataset, map_old_to_new):
"""Return an invalid dataset with one dimension renamed."""
return valid_dataset.rename(map_old_to_new)


def drop_position_var_in_ds(valid_dataset, var_name: str | list):
"""Return an invalid dataset without one or more data variable arrays."""
return valid_dataset.drop_vars(var_name)


@pytest.fixture
def not_a_dataset():
"""Return data that is not a pose tracks dataset."""
Expand All @@ -415,45 +405,37 @@ def empty_dataset():
@pytest.fixture
def missing_var_poses_dataset(valid_poses_dataset):
"""Return a poses dataset missing position variable."""
return drop_position_var_in_ds(valid_poses_dataset, var_name="position")
return valid_poses_dataset.drop_vars("position")


@pytest.fixture
def missing_var_bboxes_dataset(valid_bboxes_dataset):
"""Return a bboxes dataset missing position variable."""
return drop_position_var_in_ds(valid_bboxes_dataset, var_name="position")
return valid_bboxes_dataset.drop_vars("position")


@pytest.fixture
def missing_two_vars_bboxes_dataset(valid_bboxes_dataset):
"""Return a bboxes dataset missing position and shape variables."""
return drop_position_var_in_ds(
valid_bboxes_dataset, var_name=["position", "shape"]
)
return valid_bboxes_dataset.drop_vars(["position", "shape"])


@pytest.fixture
def missing_dim_poses_dataset(valid_poses_dataset):
"""Return a poses dataset missing the time dimension."""
return rename_dimension_in_ds(
valid_poses_dataset, map_old_to_new={"time": "tame"}
)
return valid_poses_dataset.rename({"time": "tame"})


@pytest.fixture
def missing_dim_bboxes_dataset(valid_bboxes_dataset):
"""Return a bboxes dataset missing the time dimension."""
return rename_dimension_in_ds(
valid_bboxes_dataset, map_old_to_new={"time": "tame"}
)
return valid_bboxes_dataset.rename({"time": "tame"})


@pytest.fixture
def missing_two_dims_bboxes_dataset(valid_bboxes_dataset):
"""Return a bboxes dataset missing the time and space dimensions."""
return rename_dimension_in_ds(
valid_bboxes_dataset, map_old_to_new={"time": "tame", "space": "spice"}
)
return valid_bboxes_dataset.rename({"time": "tame", "space": "spice"})


@pytest.fixture(params=["displacement", "velocity", "acceleration"])
Expand Down

0 comments on commit 4bfb237

Please sign in to comment.