From 985bbff22d2a385ee91764525af8d475e7fc5b87 Mon Sep 17 00:00:00 2001 From: Matthew Murray Date: Mon, 7 Oct 2024 16:00:27 -0700 Subject: [PATCH] address review --- .../pylibcudf/pylibcudf/tests/test_nvtext_generate_ngrams.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/pylibcudf/pylibcudf/tests/test_nvtext_generate_ngrams.py b/python/pylibcudf/pylibcudf/tests/test_nvtext_generate_ngrams.py index aedb4efddd0..5cf9874d595 100644 --- a/python/pylibcudf/pylibcudf/tests/test_nvtext_generate_ngrams.py +++ b/python/pylibcudf/pylibcudf/tests/test_nvtext_generate_ngrams.py @@ -49,3 +49,6 @@ def test_hash_character_ngrams(input_col, ngram): len(got) == max(0, len(s.as_py()) - ngram + 1) for got, s in zip(pa_result, input_col) ) + assert pa_result.type == pa.list_( + pa.field("element", pa.uint32(), nullable=False) + )