fix unit test

NeuralNetworkVerification · Aug 12, 2024 · 19fec6a · 19fec6a
1 parent 1905555
commit 19fec6a
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/src/engine/SumOfInfeasibilitiesManager.cpp b/src/engine/SumOfInfeasibilitiesManager.cpp
@@ -351,6 +351,12 @@ void SumOfInfeasibilitiesManager::setPhaseStatusInCurrentPhasePattern(
     _currentPhasePattern[constraint] = phase;
 }
 
+void SumOfInfeasibilitiesManager::setPLConstraintsInCurrentPhasePattern(
+    const Vector<PiecewiseLinearConstraint *> &constraints )
+{
+    _plConstraintsInCurrentPhasePattern = constraints;
+}
+
 void SumOfInfeasibilitiesManager::getCostReduction( PiecewiseLinearConstraint *plConstraint,
                                                     double &reducedCost,
                                                     PhaseStatus &phaseOfReducedCost ) const

diff --git a/src/engine/SumOfInfeasibilitiesManager.h b/src/engine/SumOfInfeasibilitiesManager.h
@@ -111,6 +111,8 @@ class SumOfInfeasibilitiesManager
     void setPhaseStatusInCurrentPhasePattern( PiecewiseLinearConstraint *constraint,
                                               PhaseStatus phase );
 
+    void
+    setPLConstraintsInCurrentPhasePattern( const Vector<PiecewiseLinearConstraint *> &constraints );
 
 private:
     const List<PiecewiseLinearConstraint *> &_plConstraints;

diff --git a/src/engine/tests/Test_SumOfInfeasibilitiesManager.h b/src/engine/tests/Test_SumOfInfeasibilitiesManager.h
@@ -290,6 +290,8 @@ class SumOfInfeasibilitiesManagerTestSuite : public CxxTest::TestSuite
                                       new SumOfInfeasibilitiesManager( ipq, tableau ) ) );
 
         TS_ASSERT_THROWS_NOTHING( soiManager->initializePhasePattern() );
+        TS_ASSERT_THROWS_NOTHING(
+            soiManager->setPLConstraintsInCurrentPhasePattern( plConstraints ) );
 
         for ( const auto &plConstraint : plConstraints )
         {
@@ -313,6 +315,7 @@ class SumOfInfeasibilitiesManagerTestSuite : public CxxTest::TestSuite
         TS_ASSERT_THROWS_NOTHING( plConstraints[3]->getCostFunctionComponent(
             cost1, *( plConstraints[3]->getAllCases().begin() ) ) );
 
+
         TS_ASSERT_EQUALS( soiManager->getConstraintsUpdatedInLastProposal().size(), 1u );
         TS_ASSERT_EQUALS( *soiManager->getConstraintsUpdatedInLastProposal().begin(),
                           plConstraints[1] );