Skip to content

Commit

Permalink
Update tests
Browse files Browse the repository at this point in the history
Update tests for issue #1077. We update the test name to something more
descriptive, and update the test docstring to something more
informative.
  • Loading branch information
Adam Lesnikowski committed Sep 27, 2021
1 parent a319501 commit 0be389e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/unit/loader/test_torch_dataloader.py
Original file line number Diff line number Diff line change
Expand Up @@ -512,9 +512,9 @@ def test_sparse_tensors(sparse_dense):


@pytest.mark.parametrize("pad_left", [False, True])
def test_torch_dataloader_left_padding(pad_left):
def test_sparse_tensor_left_padding(pad_left):
"""Tests the pad_left functionality of our Torch dataloader
to pad data on the left."""
to pad data on the left for sparse tensors."""
df = cudf.DataFrame({"A": [[3, 1, 5, 1], [9, 2], [6]], "B": [[3, 1, 5, 1, 9], [2], [6, 5, 3]]})
categorical_columns = ["A", "B"]
sparse_max = {"A": 5, "B": 8}
Expand Down

0 comments on commit 0be389e

Please sign in to comment.