From 0be389eed146e374bb9b27800fd24b9c01a10f78 Mon Sep 17 00:00:00 2001 From: Adam Lesnikowski Date: Mon, 27 Sep 2021 20:05:16 +0000 Subject: [PATCH] Update tests Update tests for issue #1077. We update the test name to something more descriptive, and update the test docstring to something more informative. --- tests/unit/loader/test_torch_dataloader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/loader/test_torch_dataloader.py b/tests/unit/loader/test_torch_dataloader.py index 4a00b473a86..43624f70de6 100644 --- a/tests/unit/loader/test_torch_dataloader.py +++ b/tests/unit/loader/test_torch_dataloader.py @@ -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}