Some small bug fixes

[libdai.git] / tests / testdai.cpp

diff --git a/tests/testdai.cpp b/tests/testdai.cpp
index 9b61d57..1f2b59f 100644 (file)
--- a/tests/testdai.cpp
+++ b/tests/testdai.cpp
@@ -354,6 +354,8 @@ int main( int argc, char *argv[] ) {
                         md = me;
                     if( isnan( ae ) )
                         md = ae;
+                    if( md == INFINITY )
+                        md = 1.0;
                     cout << md << "\t";
                 } else
                     cout << "N/A    \t";