Rename test Op to not confuse pytest

Author: ricardoV94

tests/tensor/test_blockwise.py

@@ -38,15 +38,15 @@ def test_vectorize_blockwise():
     assert new_vect_node.inputs[0] is tns4
 
 
-class TestOp(Op):
+class MyTestOp(Op):
     def make_node(self, *inputs):
         return Apply(self, inputs, [i.type() for i in inputs])
 
     def perform(self, *args, **kwargs):
         raise NotImplementedError("Test Op should not be present in final graph")
 
 
-test_op = TestOp()
+test_op = MyTestOp()
 
 
 def test_vectorize_node_default_signature():
@@ -56,7 +56,7 @@ def test_vectorize_node_default_signature():
 
     vect_node = vectorize_node(node, mat, mat)
     assert isinstance(vect_node.op, Blockwise) and isinstance(
-        vect_node.op.core_op, TestOp
+        vect_node.op.core_op, MyTestOp
     )
     assert vect_node.op.signature == ("(i00),(i10,i11)->(o00),(o10,o11)")
 
@@ -88,7 +88,7 @@ def test_blockwise_shape():
 
     shape_fn = pytensor.function([inp], out.shape)
     assert not any(
-        isinstance(getattr(n.op, "core_op", n.op), TestOp)
+        isinstance(getattr(n.op, "core_op", n.op), MyTestOp)
         for n in shape_fn.maker.fgraph.apply_nodes
     )
     assert tuple(shape_fn(inp_test)) == (5, 3, 4)
@@ -100,13 +100,13 @@ def test_blockwise_shape():
 
     shape_fn = pytensor.function([inp], out.shape)
     assert any(
-        isinstance(getattr(n.op, "core_op", n.op), TestOp)
+        isinstance(getattr(n.op, "core_op", n.op), MyTestOp)
         for n in shape_fn.maker.fgraph.apply_nodes
     )
 
     shape_fn = pytensor.function([inp], out.shape[:-1])
     assert not any(
-        isinstance(getattr(n.op, "core_op", n.op), TestOp)
+        isinstance(getattr(n.op, "core_op", n.op), MyTestOp)
         for n in shape_fn.maker.fgraph.apply_nodes
     )
     assert tuple(shape_fn(inp_test)) == (5, 4)
@@ -124,20 +124,20 @@ def test_blockwise_shape():
 
     shape_fn = pytensor.function([inp1, inp2], [out.shape for out in outs])
     assert any(
-        isinstance(getattr(n.op, "core_op", n.op), TestOp)
+        isinstance(getattr(n.op, "core_op", n.op), MyTestOp)
         for n in shape_fn.maker.fgraph.apply_nodes
     )
 
     shape_fn = pytensor.function([inp1, inp2], outs[0].shape)
     assert not any(
-        isinstance(getattr(n.op, "core_op", n.op), TestOp)
+        isinstance(getattr(n.op, "core_op", n.op), MyTestOp)
         for n in shape_fn.maker.fgraph.apply_nodes
     )
     assert tuple(shape_fn(inp1_test, inp2_test)) == (7, 5, 3, 4)
 
     shape_fn = pytensor.function([inp1, inp2], [outs[0].shape, outs[1].shape[:-1]])
     assert not any(
-        isinstance(getattr(n.op, "core_op", n.op), TestOp)
+        isinstance(getattr(n.op, "core_op", n.op), MyTestOp)
         for n in shape_fn.maker.fgraph.apply_nodes
     )
     assert tuple(shape_fn(inp1_test, inp2_test)[0]) == (7, 5, 3, 4)