VariableIdentifier: Less hacky way to identify sub vars, add a prefix feature

1 files changed, 13 insertions, 0 deletions

diff --git a/test/expression/expression_helper.py b/test/expression/expression_helper.py
index 3ff6f5da..e1674fdc 100644
--- a/test/expression/expression_helper.py
+++ b/test/expression/expression_helper.py
@@ -35,6 +35,19 @@ class TestExpressionExpressionHelper(unittest.TestCase):
             new_expr = new_expr.replace_expr({var_id: var_value})
         self.assertEqual(exprf, new_expr)
 
+        # Test prefix
+        vi = Variables_Identifier(exprf, var_prefix="prefix_v")
+
+        # Use __str__
+        print vi
+
+        # Test the result
+        new_expr = vi.equation
+        ## Force replace in the variable dependency order
+        for var_id, var_value in reversed(vi.vars.items()):
+            new_expr = new_expr.replace_expr({var_id: var_value})
+        self.assertEqual(exprf, new_expr)
+
 if __name__ == '__main__':
     testcase = TestExpressionExpressionHelper
     testsuite = unittest.TestLoader().loadTestsFromTestCase(testcase)