Merge pull request #18 from anexia/test_pandas

Test caching for pandas dataframe

.github/workflows/test.yml

@@ -25,7 +25,7 @@ jobs:
         python-version: ${{ matrix.python-version }}
 
     - name: Install CI dependencies
-      run: pip install flake8 codecov ${{ matrix.pytest-version }} ${{ matrix.pytest-django-version }}
+      run: pip install flake8 codecov pandas ${{ matrix.pytest-version }} ${{ matrix.pytest-django-version }}
 
     - name: Install Django ${{ matrix.django-version }}
       run: pip install 'Django==${{ matrix.django-version }}'

tests/functional/test_sync.py

@@ -23,3 +23,10 @@ def test_sync_two_requests(self, mock_func):
         self.client.get('/')
         self.assertEqual(mock_func.call_count, 4)
         self.assertEqual(cached_func.call_count, 2)
+
+    def test_sync_pandas_dataframe(self):
+        """
+        Esure that the cache_for_request decorator works with pandas dataframes.
+        """
+        response = self.client.get('/pandas_dataframe/')
+        self.assertEqual(response.status_code, 200)

tests/testapp/urls.py

@@ -1,7 +1,8 @@
 from django.urls import path
 
-from testapp.views.sync import index_view
+from testapp.views.sync import index_view, pandas_view
 
 urlpatterns = [
     path('', index_view, name='index'),
+    path('pandas_dataframe/', pandas_view, name='pandas_dataframe'),
 ]

tests/testapp/utils.py

@@ -1,3 +1,4 @@
+import pandas as pd
 from datetime import datetime
 
 from django_request_cache import cache_for_request
@@ -6,3 +7,21 @@
 @cache_for_request
 def retrieve_time():
     return datetime.now()
+
+
+def retrieve_dataframe_uncached():
+    return pd.DataFrame({
+        'a': [1, 2, 3],
+        'b': [4.0, 5.0, 6.0],
+        'c': ['a', 'b', 'c'],
+        'd': [True, False, True],
+        'e': datetime.now(),
+        'f': pd.NA
+    })
+
+
+@cache_for_request
+def retrieve_dataframe_cached():
+    return retrieve_dataframe_uncached()
+
+

tests/testapp/views/sync.py

@@ -1,8 +1,27 @@
 from django.http import HttpResponse
-from testapp.utils import retrieve_time
+from testapp.utils import retrieve_time, retrieve_dataframe_cached, retrieve_dataframe_uncached
 
 
 def index_view(request):
     retrieve_time()
     retrieve_time()
     return HttpResponse()
+
+
+def pandas_view(request):
+    """
+    Esure that the cache_for_request decorator works with pandas dataframes.
+    """
+    df1 = retrieve_dataframe_cached()
+    df2 = retrieve_dataframe_cached()
+
+    if not df1.equals(df2):
+        raise Exception("Dataframes are not equal")
+
+    df3 = retrieve_dataframe_uncached()
+    df4 = retrieve_dataframe_uncached()
+
+    if df3.equals(df4):
+        raise Exception("Dataframes are equal")
+
+    return HttpResponse()