diff --git a/tests/test_3210_to_raggedtensor_from_raggedtensor.py b/tests/test_3210_to_raggedtensor_from_raggedtensor.py index 3546279146..07b6facf64 100644 --- a/tests/test_3210_to_raggedtensor_from_raggedtensor.py +++ b/tests/test_3210_to_raggedtensor_from_raggedtensor.py @@ -111,6 +111,10 @@ def test_convert_from_raggedtensor(): result1.content.data == ak.to_backend(np_array1, result1.backend).layout.data ).all() assert (result1.offsets.data == [0, 2, 3, 3, 5]).all() + assert ( + result1.offsets.data + == ak.to_backend([0, 2, 3, 3, 5], result1.backend).layout.data + ).all() assert from_raggedtensor(tf_array1).to_list() == ak_array1.to_list() tf_array2 = tf.RaggedTensor.from_nested_row_splits(