Add BooleanAtom to LambdaDoubleVisitor

symengine · Jul 21, 2018 · b7bdadf · b7bdadf
1 parent 830c2a7
commit b7bdadf
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/symengine/lambda_double.h b/symengine/lambda_double.h
@@ -501,11 +501,17 @@ class LambdaRealDoubleVisitor
                                             // a true boolean
             };
         } else {
-            throw SymEngineException("LambdaDoubleVisitor only ``Interval`` "
+            throw SymEngineException("LambdaDoubleVisitor: only ``Interval`` "
                                      "implemented for ``Contains``.");
         }
     }
 
+    void bvisit(const BooleanAtom &ba)
+    {
+        const bool val = ba.get_val();
+        result_ = [=](const double * /* x */) { return (val) ? 1.0 : 0.0; };
+    }
+
     void bvisit(const Piecewise &pw)
     {
         if (neq(*pw.get_vec().back().second, *boolTrue)) {

diff --git a/symengine/tests/eval/test_lambda_double.cpp b/symengine/tests/eval/test_lambda_double.cpp
@@ -113,8 +113,8 @@ TEST_CASE("Evaluate to double", "[lambda_double]")
     REQUIRE(::fabs(d - 1.1) < 1e-12);
     d = v.call({2.2, 3.3});
     REQUIRE(::fabs(d - 3.3) < 1e-12);
-    d = v.call({2.2, 5.5});
-    REQUIRE(::fabs(d - 7.7) < 1e-12);
+    d = v.call({5.5, 3.3});
+    REQUIRE(::fabs(d - 8.8) < 1e-12);
 }
 
 TEST_CASE("Evaluate double cse", "[lambda_double_cse]")