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.boolean_eval
import stl.robustness
import stl.smooth_robustness
import pandas as pd
from nose2.tools import params
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):
@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)
stl_eval = stl.robustness.pointwise_robustness(phi)
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