As a followup of Spark-36814, to create a test to extend ColumnarBatch to prevent future changes to break it.
[Github] Pull Request #34087 (flyrain)