    def test_data_collection_batch_inter(self):
        """
        target: test batch() API for DataCollection 
        method: create an iter data collection and batch it
        expected: return batch successfully
        """
        data_size = 6
        data = iter(range(data_size))
        dc = towhee.dc(data)
        size = 3
        result = [list(batch) for batch in dc.batch(size, drop_tail=True)]
        assert len(result) == int(data_size/size) 

        return True
target: test batch() API for DataCollection
method: create an iter data collection and batch it
expected: return batch successfully
tests/testcases/test_data_collection.py