fix bugs in binary search (still broken)

This commit is contained in:
Marcell Vazquez-Chanlatte 2016-10-09 22:34:28 -07:00
parent 28f755edc5
commit c6810d8d75
2 changed files with 19 additions and 19 deletions

View file

@ -7,24 +7,23 @@ from sympy import Symbol
oo = float('inf')
ex1 = ("A > a?", ("a?",), {"a?": (0, 10)}, {"a?": True}, {"a?": 1})
ex1 = ("F[0, b?](A > a?)", ("a?", "b?"), {"a?": (0, 10), "b?": (0, 10)},
{"a?": True, "b?": False}, {"a?": 4, "b?": 0.2})
ex1 = ("A > a?", ("a?",), {"a?": (0, 10)}, {"a?": False}, {"a?": 1})
ex2 = ("F[0, b?](A > a?)", ("a?", "b?"), {"a?": (0, 10), "b?": (0, 5)},
{"a?": False, "b?": True}, {"a?": 4, "b?": 0.2})
x = pd.DataFrame([[1,2], [1,4], [4,2]], index=[0,0.1,0.2],
columns=["A", "B"])
class TestSTLRobustness(unittest.TestCase):
@params(ex1)
@params(ex1, ex2)
def test_lex_synth(self, phi_str, order, ranges, polarity, val):
phi = stl.parse(phi_str)
val2 = stl.robustness.lex_param_project(
phi, x, order=order, ranges=ranges, polarity=polarity)
phi = stl.robustness.set_params(phi, val)
phi = stl.robustness.set_params(phi, val2)
stl_eval = stl.robustness.pointwise_robustness(phi)
self.assertAlmostEqual(stl_eval(x, 0), 0)
self.assertAlmostEqual(stl_eval(x, 0), 0, delta=0.01)
for var in order:
self.assertAlmostEqual(val2[var], val[var], delta=0.01)