Changeset 831 for mystic/tests/test_symbolic.py

Timestamp:

09/28/15 20:45:46 (8 months ago)

Author:

mmckerns

Message:

add simplify to simplify equations for a single variable on the LHS;
use simplify in many examples where symbolic equations are given;
add comparator function, to determine the comparator ('<=', '=', etc);
add test_simplify to test_symbolic

File:

• mystic/tests/test_symbolic.py (modified) (2 diffs)

mystic/tests/test_symbolic.py

@@ -76 +76 @@
   assert (x[1] - x[0]) - 1.0 == mean(x[:-1])
 
+def test_simplify():
+  constraints = """
+  mean([x0, x1, x2]) <= 5.0
+  x0 <= x1 + x2"""
+
+  from mystic.math.measures import mean
+  _constraints = simplify(constraints)
+  solv = generate_solvers(_constraints)
+  constraint = generate_constraint(solv)
+  x = constraint([1.0, -2.0, -3.0])
+  assert all(x) == all([-5.0, -2.0, -3.0])
+  assert mean(x) <= 5.0
+  assert x[0] <= x[1] + x[2]
+
 
 if __name__ == '__main__':
@@ -82 +96 @@
   test_generate_constraint()
   test_solve_constraint()
+  test_simplify()