From 452bedd4236329b6900eb7713c2d59c777656b71 Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Fri, 4 Nov 2022 08:08:29 +0100
Subject: [PATCH] Two more tests
---
pystencils_tests/test_sympyextensions.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/pystencils_tests/test_sympyextensions.py b/pystencils_tests/test_sympyextensions.py
index b0ddd6967..1929cc066 100644
--- a/pystencils_tests/test_sympyextensions.py
+++ b/pystencils_tests/test_sympyextensions.py
@@ -177,6 +177,17 @@ def test_count_operations():
ops = count_operations(expr, only_type=None)
assert ops['muls'] == 99
+ expr = DivFunc(1, sp.UnevaluatedExpr(sp.Mul(*[x]*100, evaluate=False)))
+ ops = count_operations(expr, only_type=None)
+ assert ops['divs'] == 1
+ assert ops['muls'] == 99
+
+ expr = DivFunc(y + z, sp.UnevaluatedExpr(sp.Mul(*[x]*100, evaluate=False)))
+ ops = count_operations(expr, only_type=None)
+ assert ops['adds'] == 1
+ assert ops['divs'] == 1
+ assert ops['muls'] == 99
+
def test_common_denominator():
x = sympy.symbols('x')
--
GitLab