diff --git a/stl/robustness.py b/stl/robustness.py index 058dd0a..c9475c0 100644 --- a/stl/robustness.py +++ b/stl/robustness.py @@ -41,7 +41,7 @@ def _(stl): @pointwise_robustness.register(stl.Neg) def _(stl): - return lambda x, t: -pointwise_robustness(arg)(x, t) + return lambda x, t: -pointwise_robustness(stl.arg)(x, t) op_lookup = { diff --git a/stl/test_robustness.py b/stl/test_robustness.py index c984c7e..730e878 100644 --- a/stl/test_robustness.py +++ b/stl/test_robustness.py @@ -32,8 +32,10 @@ class TestSTLRobustness(unittest.TestCase): @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) + r1 = stl.robustness.pointwise_robustness(phi)(x, 0) + r2 = stl.robustness.pointwise_robustness(~phi)(x, 0) + self.assertEqual(r1, r) + self.assertEqual(r1, -r2) @params(ex1, ex2, ex3, ex4, ex5, ex6)