fix: solve confusion between parts and size by setting a non squared example

Author: 12rambau

geetools/ee_list.py

@@ -319,14 +319,14 @@ def chunked(self, parts: int | ee.Number) -> ee.List:
         idxList = ee.List.sequence(0, parts.subtract(1))
 
         def compute_parts(idx: ee.Number) -> ee.List:
-            start = ee.Number(idx).multiply(parts)
+            start = ee.Number(idx).multiply(size)
             return self._obj.slice(start, start.add(size))
 
         chunked = ee.List(idxList.map(compute_parts))
 
         # add the rest of the elements (the rest of the euclidean division)
         # to the last chunk of the list.
-        rest = self._obj.slice(totalSize.subtract(totalSize.mod(size)), totalSize)
+        rest = self._obj.slice(totalSize.subtract(totalSize.mod(parts)), totalSize)
         lastChunk = ee.List(chunked.get(-1)).cat(rest)
 
         return chunked.set(-1, lastChunk)

tests/test_List.py

@@ -120,13 +120,13 @@ class TestChunked:
     """Test the chunked method."""
 
     def test_chunked_even(self, ee_list_regression):
-        input = ee.List([1, 2, 3, 4, 5, 6, 7, 8, 9])
-        chunked = input.geetools.chunked(3)
-        # should return -> [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
+        input = ee.List([1, 2, 3, 4, 5, 6, 7, 8])
+        chunked = input.geetools.chunked(4)
+        # should return -> [[1, 2], [3, 4], [5, 6], [7, 8]]
         ee_list_regression.check(chunked)
 
     def test_chunked_odd(self, ee_list_regression):
-        input = ee.List([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
-        chunked = input.geetools.chunked(3)
-        # should return -> [[1, 2, 3], [4, 5, 6], [7, 8, 9, 10]]
+        input = ee.List([1, 2, 3, 4, 5, 6, 7, 8, 9])
+        chunked = input.geetools.chunked(4)
+        # should return -> [[1, 2], [3, 4], [5, 6], [7, 8, 9]]
         ee_list_regression.check(chunked)

tests/test_List/serialized_test_chunked_even.yml

@@ -1,9 +1,8 @@
 - - 1
   - 2
-  - 3
-- - 4
-  - 5
+- - 3
+  - 4
+- - 5
   - 6
 - - 7
   - 8
-  - 9

tests/test_List/serialized_test_chunked_odd.yml

@@ -1,10 +1,9 @@
 - - 1
   - 2
-  - 3
-- - 4
-  - 5
+- - 3
+  - 4
+- - 5
   - 6
 - - 7
   - 8
   - 9
-  - 10