Merge pull request #274 from seanmacavaney/is_transformer_bug

fixed bug with is_transformer

pyterrier/transformer.py

@@ -16,7 +16,7 @@ def is_function(v):
     return isinstance(v, types.FunctionType)
 
 def is_transformer(v):
-    if isinstance(v, TransformerBase):
+    if isinstance(v, Transformer):
         return True
     return False
 

tests/test_transformer.py

@@ -0,0 +1,19 @@
+import pyterrier as pt
+import unittest
+from .base import BaseTestCase
+import os
+import pandas as pd
+
+class TestTransformer(BaseTestCase):
+
+    def test_is_transformer(self):
+        class MyTransformer1(pt.Transformer):
+            pass
+        class MyTransformer2(pt.transformer.TransformerBase):
+            pass
+        class MyTransformer3(pt.transformer.IterDictIndexerBase):
+            pass
+        class MyTransformer4(pt.transformer.EstimatorBase):
+            pass
+        for T in [MyTransformer1, MyTransformer2, MyTransformer3, MyTransformer4]:
+            self.assertTrue(pt.transformer.is_transformer(T()))