From 900da9e556385c6af7b3f14890abc139b6900a26 Mon Sep 17 00:00:00 2001
From: Tim Quatmann <tim.quatmann@cs.rwth-aachen.de>
Date: Mon, 29 Jul 2019 16:56:19 +0200
Subject: [PATCH] Fixed EndComponentEliminatorTest

---
 src/test/storm/transformer/EndComponentEliminatorTest.cpp | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/test/storm/transformer/EndComponentEliminatorTest.cpp b/src/test/storm/transformer/EndComponentEliminatorTest.cpp
index 087310183..3dd19899f 100644
--- a/src/test/storm/transformer/EndComponentEliminatorTest.cpp
+++ b/src/test/storm/transformer/EndComponentEliminatorTest.cpp
@@ -94,8 +94,12 @@ TEST(NeutralECRemover, SimpleModelTest) {
     for (uint64_t oldState = 0; oldState < expectedOldToNewStateMapping.size(); ++oldState) {
         uint64_t expectedNewState = expectedOldToNewStateMapping[oldState];
         uint64_t actualNewState = res.oldToNewStateMapping[oldState];
-        ASSERT_EQ(actualNewState < std::numeric_limits<uint_fast64_t>::max(), expectedNewState < std::numeric_limits<uint_fast64_t>::max()) << " Mapping does not match for oldState " << oldState;
-        actualToExpectedStateMapping[actualNewState] = expectedNewState;
+        if (actualNewState < std::numeric_limits<uint_fast64_t>::max()) {
+            ASSERT_LT(expectedNewState, std::numeric_limits<uint_fast64_t>::max()) << " Mapping does not match for oldState " << oldState;
+            actualToExpectedStateMapping[actualNewState] = expectedNewState;
+        } else {
+            ASSERT_LT(expectedNewState, actualNewState) << " Mapping does not match for oldState " << oldState;
+        }
     }
     std::vector<uint64_t> actualToExpectedRowMapping;
     for (uint64_t actualRow = 0; actualRow < res.matrix.getRowCount(); ++actualRow) {