Skip to content

Commit

Permalink
Update expected_dataset docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
lochhh committed Feb 5, 2024
1 parent 28bab9f commit 9445de3
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions tests/test_unit/test_kinematics.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,12 @@ def expected_dataarray(self, valid_pose_dataset):

@pytest.fixture
def expected_dataset(self, valid_pose_dataset):
"""Return a function to generate an expected dataset."""
"""Return a function to generate the expected dataset
for different kinematic properties."""

def _expected_dataset(name):
"""Return an xarray.Dataset with expected norm and
theta values."""
dims = valid_pose_dataset.pose_tracks.dims[:-1]
ds = xr.Dataset(
data_vars={
Expand All @@ -49,7 +52,6 @@ def _expected_dataset(name):
# Set all values to zero
ds.norm[:] = 0
ds.theta[:] = 0
ds.attrs["name"] = name
return ds

return _expected_dataset
Expand Down

0 comments on commit 9445de3

Please sign in to comment.