From 60937677723c542d73c87939f045aaacd200ffa9 Mon Sep 17 00:00:00 2001 From: Marcell Vazquez-Chanlatte Date: Tue, 28 Feb 2017 13:35:01 -0800 Subject: [PATCH] fix negation bug in robustness --- stl/robustness.py | 2 +- stl/test_robustness.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) 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)