dataset.test.py

@@ -3,7 +3,7 @@ import unittest
 
 class Test(unittest.TestCase):
     def test_padding_array(self):
-        self.assertEqual(padding_array([[1,2],[3]]),[[1,2],[3,0]])
+        self.assertEquals(padding_array([[1,2],[3]]),[[1,2],[3,0]])
 
 if __name__ == '__main__':
     unittest.main()

groupby_index.py

@@ -1,35 +0,0 @@
-import itertools
-from typing import Tuple, TypeVar, Iterable
-
-T = TypeVar('T')
-
-def groupby_index(iter: Iterable[T],n:int) -> Iterable[Iterable[T]]:
-    """group list by index
-
-    Args:
-        iter (Iterable[T]): iterator to group by index
-        n (int): The size of groups
-
-    Returns:
-        Iterable[Iterable[T]]: iterable object to group by index
-
-    >>> [*map(lambda x:[*x],groupby_index([1,2,3,4],2))]
-    [[1, 2], [3, 4]]
-    """
-    def keyfunc(x: Tuple[int,T]) -> int:
-        k, _ = x
-        return (k // n)
-    def mapper(x: Tuple[int, Tuple[int, T]]):
-        _, v = x
-        return map(lambda y: y[1],v)
-    g = itertools.groupby(enumerate(iter), keyfunc)
-    return map(mapper,g)
-
-if __name__ == "__main__":
-    test_list = [*range(20,-10,-1)]
-    for g in groupby_index(test_list,4):
-        print([*g])
-    print([*map(lambda x:[*x],groupby_index([1,2,3,4],2))])
-    for g in groupby_index([1,2,3,4],2):
-        print([*g])
-    

groupby_index.test.py

@@ -1,9 +0,0 @@
-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()