Merge pull request #507 from ddobrinskiy/patch/00_test

extend test_eq_type

Author: jph00

fastcore/__init__.py

@@ -1 +1 @@
-__version__ = "1.5.28"
+__version__ = "1.5.29"

fastcore/test.py

@@ -41,7 +41,8 @@ def test_eq_type(a,b):
     "`test` that `a==b` and are same type"
     test_eq(a,b)
     test_eq(type(a),type(b))
-    if isinstance(a,(list,tuple)): test_eq(map(type,a),map(type,b))
+    if isinstance(a,(list,tuple)): test_eq(map(type,a),map(type,b)) # type of each element
+    if isinstance(a, (torch.Tensor, pd.Series, np.ndarray)): test_eq(a.dtype, b.dtype) # dtypes of both tensors
 
 # %% ../nbs/00_test.ipynb 27
 def test_ne(a,b):

nbs/00_test.ipynb

@@ -339,7 +339,8 @@
     "    \"`test` that `a==b` and are same type\"\n",
     "    test_eq(a,b)\n",
     "    test_eq(type(a),type(b))\n",
-    "    if isinstance(a,(list,tuple)): test_eq(map(type,a),map(type,b))"
+    "    if isinstance(a,(list,tuple)): test_eq(map(type,a),map(type,b)) # type of each element\n",
+    "    if isinstance(a, (torch.Tensor, pd.Series, np.ndarray)): test_eq(a.dtype, b.dtype) # dtypes of both tensors"
    ]
   },
   {
@@ -354,7 +355,10 @@
     "test_fail(lambda: test_eq_type(1,1.))\n",
     "test_eq_type([1,1],[1,1])\n",
     "test_fail(lambda: test_eq_type([1,1],(1,1)))\n",
-    "test_fail(lambda: test_eq_type([1,1],[1,1.]))"
+    "test_fail(lambda: test_eq_type([1,1],[1,1.]))\n",
+    "test_fail(lambda: test_eq_type(torch.zeros(10), torch.zeros(10, dtype=torch.float64)))\n",
+    "test_fail(lambda: test_eq_type(torch.zeros(10), np.zeros(10)))\n",
+    "test_eq_type(torch.zeros(3), torch.Tensor([0, 0, 0]))"
    ]
   },
   {

settings.ini

@@ -8,7 +8,7 @@ author = Jeremy Howard and Sylvain Gugger
 author_email = infos@fast.ai
 copyright = fast.ai
 branch = master
-version = 1.5.28
+version = 1.5.29
 min_python = 3.7
 audience = Developers
 language = English