Test: support for aggregation syntax in pandas < 0.25

Author: aisipos

django_pandas/managers.py

@@ -241,7 +241,7 @@ def to_timeseries(self, fieldnames=(), verbose=True,
 
         if freq is not None:
             if agg_kwargs is None:
-                agg_kwargs=dict(func='mean')
+                agg_kwargs=dict()
             if agg_args is None:
                 agg_args=[]
             df = df.resample(freq, **rs_kwargs).agg(*agg_args, **agg_kwargs)

django_pandas/tests/test_manager.py

@@ -8,7 +8,9 @@
     LongTimeSeries, PivotData, Dude, Car, Spot
 )
 import pandas.util.testing as tm
+import semver
 
+PANDAS_VERSIONINFO = semver.VersionInfo.parse(pd.__version__)
 
 class DataFrameTest(TestCase):
 
@@ -120,10 +122,16 @@ def test_longstorage(self):
     def test_resampling(self):
         qs = LongTimeSeries.objects.all()
         rs_kwargs = {'kind': 'period'}
-        agg_kwargs = {'func': 'sum'}
+        agg_args = None
+        agg_kwargs = None
+        if PANDAS_VERSIONINFO >= '0.25.0':
+            agg_kwargs = {'func': 'sum'}
+        else:
+            agg_args= ['sum']
         df = qs.to_timeseries(index='date_ix', pivot_columns='series_name',
                               values='value', storage='long',
                               freq='M', rs_kwargs=rs_kwargs,
+                              agg_args=agg_args,
                               agg_kwargs=agg_kwargs)
 
         self.assertEqual([d.month for d in qs.dates('date_ix', 'month')],
@@ -135,7 +143,9 @@ def test_resampling(self):
         qs2 = WideTimeSeries.objects.all()
 
         df1 = qs2.to_timeseries(index='date_ix', storage='wide',
-                                freq='M', rs_kwargs=rs_kwargs)
+                                freq='M', rs_kwargs=rs_kwargs,
+                                agg_args=agg_args,
+                                agg_kwargs = agg_kwargs)
 
         self.assertEqual([d.month for d in qs.dates('date_ix', 'month')],
                          df1.index.month.tolist())

setup.py

@@ -42,6 +42,7 @@
     tests_require=[
         "pandas>=0.20.1",
         "coverage>=4.0",
+        "semver==2.10.1"
                    ],
     test_suite="runtests.runtests"