From 27da05517448d3ccf16f483896f3587e751f43e4 Mon Sep 17 00:00:00 2001
From: Alexander Reinauer <areinauer@icp.uni-stuttgart.de>
Date: Sun, 21 Apr 2024 14:23:52 +0200
Subject: [PATCH] Fix new_merged for AssignmentCollections
---
tests/test_assignment_collection.py | 16 ++++++++++++++++
1 file changed, 16 insertions(+)
diff --git a/tests/test_assignment_collection.py b/tests/test_assignment_collection.py
index ced0a36e1..797465eca 100644
--- a/tests/test_assignment_collection.py
+++ b/tests/test_assignment_collection.py
@@ -170,3 +170,19 @@ def test_new_merged():
assert ps.Assignment(d[0, 0](0), sp.symbols('xi_0')) in merged_ac.main_assignments
assert a1 in merged_ac.subexpressions
assert a3 in merged_ac.subexpressions
+
+ a1 = ps.Assignment(a, 20)
+ a2 = ps.Assignment(a, 10)
+ acommon = ps.Assignment(b, a)
+
+ # main assignments
+ a3 = ps.Assignment(f[0, 0](0), b)
+ a4 = ps.Assignment(d[0, 0](0), b)
+
+ ac = ps.AssignmentCollection([a3], subexpressions=[a1, acommon])
+ ac2 = ps.AssignmentCollection([a4], subexpressions=[a2, acommon])
+
+ merged_ac = ac.new_merged(ac2).new_without_subexpressions()
+
+ assert ps.Assignment(f[0, 0](0), 20) in merged_ac.main_assignments
+ assert ps.Assignment(d[0, 0](0), 10) in merged_ac.main_assignments
--
GitLab