new class ExerciseFunctionPandas

demo-notebooks/exercise07-miscell.py

@@ -140,5 +140,11 @@ def longargs(x):
 # %%
 exo_longargs.correction(longargs)
 
+# %% [markdown] cell_style="center"
+# # pandas
+
+# %% [markdown]
+# As of jan 2021, the new class `ExerciseFunction` allows to deal with funtions that return a pandas `DataFrame` or `Series` - a demo sample would be great
+
 # %% [markdown]
 # ***

nbautoeval/__init__.py

@@ -2,10 +2,12 @@
 from .renderer import Renderer, PPrintRenderer, MultilineRenderer, ImshowRenderer
 from .callrenderer import CallRenderer, PPrintCallRenderer, IsliceRenderer
 
-from .exercise_function import ExerciseFunction, ExerciseFunctionNumpy
+from .exercise_function import (
+    ExerciseFunction, ExerciseFunctionNumpy, ExerciseFunctionPandas)
 from .exercise_regexp import ExerciseRegexp, ExerciseRegexpGroups
 from .exercise_generator import ExerciseGenerator
-from .exercise_class import ExerciseClass, ClassScenario, ClassExpression, ClassStatement
+from .exercise_class import (
+    ExerciseClass, ClassScenario, ClassExpression, ClassStatement)
 
 from .content import (TextContent, CodeContent, MathContent,
                       MarkdownContent, MarkdownMathContent)

nbautoeval/exercise_function.py

@@ -334,6 +334,10 @@ def validate(self, expected, result):               # pylint: disable=r0201
 
         return expected == result
 
+
+
+# ExerciseFunctionNumpy - no hard requirement on pandas
+
 # see this question on SO
 # https://stackoverflow.com/questions/40659212/futurewarning-elementwise-comparison-failed-returning-scalar-but-in-the-futur
 
@@ -375,6 +379,26 @@ def validate(self, expected, result):
                     print("OOPS2", type(exc), exc)
                     return False
 
-except Exception:
-    #print("ExerciseFunctionNumpy not defined ; numpy not installed ? ")
-    pass
+except ModuleNotFoundError:
+    class ExerciseFunctionNumpy(ExerciseFunction):
+        def __init__(self, *args, **kwds):
+            print("WARNING: dummy ExerciseFunctionNumpy - numpy not installed ?")
+            super().__init__(*args, **kwds)
+
+
+# ExerciseFunctionPandas - no hard requirement on pandas
+try:
+    import pandas as pd
+
+    class ExerciseFunctionPandas(ExerciseFunction):
+        """
+        This is suitable for functions that are expected to return 
+        either a pandas DataFrame or Series object
+        """
+        def validate(self, expected, result):
+            return expected.equals(result)
+except ModuleNotFoundError:
+    class ExerciseFunctionPandas(ExerciseFunction):
+        def __init__(self, *args, **kwds):
+            print("WARNING: dummy ExerciseFunctionNumpy - numpy not installed ?")
+            super().__init__(*args, **kwds)