check for deepcopy or view rather than string

Author: dehorsley

tests/sandbox/linalg/test_linalg.py

@@ -4,6 +4,7 @@
 import pytensor
 from pytensor import function
 from pytensor import tensor as at
+from pytensor.compile import DeepCopyOp, ViewOp
 from pytensor.configdefaults import config
 from pytensor.sandbox.linalg.ops import inv_as_solve, spectral_radius_bound
 from pytensor.tensor.elemwise import DimShuffle
@@ -163,7 +164,11 @@ def test_cholesky_dot_lower():
     C = cholesky_lower(L.dot(L.T))
     f = pytensor.function([L], C)
     if config.mode != "FAST_COMPILE":
-        assert f.maker.fgraph.outputs[0].name == "L"
+        assert (f.maker.fgraph.outputs[0] == f.maker.fgraph.inputs[0]) or (
+            (o := f.maker.fgraph.outputs[0].owner)
+            and isinstance(o.op, (DeepCopyOp, ViewOp))
+            and o.inputs[0] == f.maker.fgraph.inputs[0]
+        )
 
 
 def test_cholesky_dot_upper():
@@ -175,4 +180,8 @@ def test_cholesky_dot_upper():
     C = cholesky_upper(U.T.dot(U))
     f = pytensor.function([U], C)
     if config.mode != "FAST_COMPILE":
-        assert f.maker.fgraph.outputs[0].name == "U"
+        assert (f.maker.fgraph.outputs[0] == f.maker.fgraph.inputs[0]) or (
+            (o := f.maker.fgraph.outputs[0].owner)
+            and isinstance(o.op, (DeepCopyOp, ViewOp))
+            and o.inputs[0] == f.maker.fgraph.inputs[0]
+        )