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

fix negation bug in robustness

Browse source
This commit is contained in:
Marcell Vazquez-Chanlatte 2017-02-28 13:35:01 -08:00
parent a8e84a3761
commit 6093767772
2 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
stl/robustness.py
View file

@ -41,7 +41,7 @@ def _(stl):
@pointwise_robustness.register(stl.Neg) @pointwise_robustness.register(stl.Neg)
def _(stl): def _(stl):
return lambda x, t: -pointwise_robustness(arg)(x, t) return lambda x, t: -pointwise_robustness(stl.arg)(x, t)
op_lookup = { op_lookup = {

6
stl/test_robustness.py
View file

@ -32,8 +32,10 @@ class TestSTLRobustness(unittest.TestCase):
@params(ex1, ex2, ex3, ex4, ex5, ex6) @params(ex1, ex2, ex3, ex4, ex5, ex6)
def test_robustness_value(self, phi_str, r): 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) r1 = stl.robustness.pointwise_robustness(phi)(x, 0)
self.assertEqual(stl_eval(x, 0), r) r2 = stl.robustness.pointwise_robustness(~phi)(x, 0)
self.assertEqual(r1, r)
self.assertEqual(r1, -r2)
@params(ex1, ex2, ex3, ex4, ex5, ex6) @params(ex1, ex2, ex3, ex4, ex5, ex6)