ner-study/groupby_index.test.py

import unittest
from groupby_index import *

class Test(unittest.TestCase):
    def test_padding_array(self):
        self.assertEqual([*map(lambda x:[*x],groupby_index([1,2,3,4],2))],[[1,2],[3,4]])

if __name__ == '__main__':
    unittest.main()