Fix rolling_groupby and expanding_groupby for pandas 1.3.0

nalepae · nalepae · commit f4143c5b4692 · 2022-03-15T10:45:43.000+01:00
GitHub ticket #172
diff --git a/pandarallel/data_types/expanding_groupby.py b/pandarallel/data_types/expanding_groupby.py
@@ -24,7 +24,7 @@ def get_chunks(
 
             iterator = (
                 iter(data._groupby)
-                if pandas_version <= (1, 3)
+                if pandas_version < (1, 3)
                 else data._grouper.get_iterator(data.obj)
             )
 
diff --git a/pandarallel/data_types/rolling_groupby.py b/pandarallel/data_types/rolling_groupby.py
@@ -24,7 +24,7 @@ def get_chunks(
 
             iterator = (
                 iter(data._groupby)
-                if pandas_version <= (1, 3)
+                if pandas_version < (1, 3)
                 else data._grouper.get_iterator(data.obj)
             )
 
diff --git a/tests.sh b/tests.sh
@@ -23,12 +23,11 @@ echo
 pip install pandas==1.2.0 --quiet
 pytest
 
-# TODO: Fix tests for this version of Pandas
-# echo "Pandas 1.3.0"
-# echo "------------"
-# echo
-# pip install pandas==1.3.0 --quiet
-# pytest
+echo "Pandas 1.3.0"
+echo "------------"
+echo
+pip install pandas==1.3.0 --quiet
+pytest
 
 echo "Pandas 1.4.0"
 echo "------------"

Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@ def get_chunks(`
`24`	`24`
`25`	`25`	`iterator = (`
`26`	`26`	`iter(data._groupby)`
`27`		`- if pandas_version <= (1, 3)`
	`27`	`+ if pandas_version < (1, 3)`
`28`	`28`	`else data._grouper.get_iterator(data.obj)`
`29`	`29`	`)`
`30`	`30`