TST: Add tests for log_diff_exp

aesara-devs · Jan 20, 2023 · a05de4a · a05de4a
1 parent 0091f18
commit a05de4a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/tensor/test_math.py b/tests/tensor/test_math.py
@@ -3521,3 +3521,16 @@ def test_infer_shape(self):
                 [x1, x2],
                 self.op_class,
             )
+
+def test_logdiffexp():
+    x = fmatrix()
+    y = fmatrix()
+    f = function([x, y], log(exp(x) - exp(y)))
+    graph = f.maker.fgraph.toposort()
+    ops_graph = [
+        node
+        for node in graph
+        if isinstance(node.op, Elemwise)
+        and isinstance(node.op.scalar_op, aes.Exp)
+    ]
+    assert len(ops_graph) == 0