teaching/mtl-aas
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

sketch out tests for smooth robustness

Browse source
This commit is contained in:
Marcell Vazquez-Chanlatte 2016-12-21 19:18:01 -08:00
parent 0261bb434f
commit 014110cf90
2 changed files with 31 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

31
stl/test_robustness.py
View file

@ -1,5 +1,7 @@
import stl import stl
import stl.boolean_eval
import stl.robustness import stl.robustness
import stl.smooth_robustness
import pandas as pd import pandas as pd
from nose2.tools import params from nose2.tools import params
import unittest import unittest
@ -19,7 +21,34 @@ x = pd.DataFrame([[1,2], [1,4], [4,2]], index=[0,0.1,0.2],
class TestSTLRobustness(unittest.TestCase): class TestSTLRobustness(unittest.TestCase):
@params(ex1, ex2, ex3, ex4, ex5, ex6) @params(ex1, ex2, ex3, ex4, ex5, ex6)
def test_stl(self, phi_str, r): def test_robustness_sign(self, phi_str, _):
phi = stl.parse(phi_str)
stl_eval = stl.boolean_eval.pointwise_sat(phi)
stl_eval2 = stl.robustness.pointwise_robustness(phi)
r = stl_eval2(x, 0)
assert (r == 0 or ((r > 0) == stl_eval(x, 0)))
@params(ex1, ex2, ex3, ex4, ex5, ex6)
def test_robustness_value(self, phi_str, r):
phi = stl.parse(phi_str) phi = stl.parse(phi_str)
stl_eval = stl.robustness.pointwise_robustness(phi) stl_eval = stl.robustness.pointwise_robustness(phi)
self.assertEqual(stl_eval(x, 0), r) self.assertEqual(stl_eval(x, 0), r)
@params(ex1, ex2, ex3, ex4, ex5, ex6)
def test_eps_robustness(self, phi_str, r):
phi = stl.parse(phi_str)
r = stl.robustness.pointwise_robustness(phi)(x, 0)
lo, hi = stl.smooth_robustness.smooth_robustness(phi, L=1, eps=0.1)
# hi - lo <= eps
# lo <= r <= hi
raise NotImplementedError
@params(ex1, ex2, ex3, ex4, ex5, ex6)
def test_interval_polarity(self, phi_str, r):
phi = stl.parse(phi_str)
lo, hi = stl.smooth_robustness.smooth_robustness(phi, L=1, eps=0.1)
# hi - lo > 0
raise NotImplementedError

2
stl/utils.py
View file

@ -66,7 +66,7 @@ def terms_lens(phi:"STL", bind=True) -> lens:
def param_lens(phi): def param_lens(phi):
is_sym = lambda x: isinstance(x, sympy.Symbol) is_sym = lambda x: isinstance(x, sympy.Symbol)
def focus_lens(leaf): def focus_lens(leaf):
return [lens().const, lens().id] if isinstance(leaf, LinEq) else [lens().interval[0], lens().interval[1]] return [lens().const] if isinstance(leaf, LinEq) else [lens().interval[0], lens().interval[1]]
return ast_lens(phi, pred=type_pred(LinEq, F, G), return ast_lens(phi, pred=type_pred(LinEq, F, G),
focus_lens=focus_lens).filter_(is_sym) focus_lens=focus_lens).filter_(is_sym)