Merge pull request #188 from kessido/fix_pandas_test_eq

Fix pandas test eq

Author: jph00

fastcore/imports.py

@@ -51,6 +51,8 @@ def array_equal(a,b):
     if hasattr(b, '__array__'): b = b.__array__()
     return (a==b).all()
 
+def df_equal(a,b): return a.equals(b) if isinstance_str(a, 'NDFrame') else b.equals(a)
+
 def equals(a,b):
     "Compares `a` and `b` for equality; supports sublists, tensors and arrays too"
     if (a is None) ^ (b is None): return False
@@ -59,6 +61,7 @@ def equals(a,b):
     if hasattr(b, '__array_eq__'): return b.__array_eq__(a)
     cmp = (array_equal   if isinstance_str(a, 'ndarray') or isinstance_str(b, 'ndarray') else
            array_equal   if isinstance_str(a, 'Tensor')  or isinstance_str(b, 'Tensor') else
+           df_equal      if isinstance_str(a, 'NDFrame') or isinstance_str(b, 'NDFrame') else
            operator.eq   if any_is_instance((str,dict,set), a, b) else
            all_equal     if is_iter(a) or is_iter(b) else
            operator.eq)