From 33189681bf13f88a1e39383a86079fc001ec619d Mon Sep 17 00:00:00 2001 From: Camille Mougey Date: Tue, 10 Feb 2015 09:57:50 +0100 Subject: VariableIdentifier: Less hacky way to identify sub vars, add a prefix feature --- test/expression/expression_helper.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'test/expression/expression_helper.py') 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) -- cgit 1.4.1