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) + )