TEST: Test MultiResolutionPDEDeformableRegistration::SetNumberOfItera…

…tions
Leengit · Jan 23, 2025 · f2212e9 · f2212e9
1 parent e4e10a1
commit f2212e9
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/Modules/Registration/PDEDeformable/include/itkMultiResolutionPDEDeformableRegistration.hxx b/Modules/Registration/PDEDeformable/include/itkMultiResolutionPDEDeformableRegistration.hxx
@@ -205,7 +205,7 @@ MultiResolutionPDEDeformableRegistration<TFixedImage,
 {
   // In case numberOfIterations.GetSize() differs from this->m_numberOfLevels, first set the
   // latter to equal the former.
-  SetNumberOfLevels(numberOfIterations.GetSize());
+  this->SetNumberOfLevels(numberOfIterations.GetSize());
   // Now do the requested set
   if (this->m_NumberOfIterations != numberOfIterations)
   {

diff --git a/Modules/Registration/PDEDeformable/test/itkMultiResolutionPDEDeformableRegistrationTest.cxx b/Modules/Registration/PDEDeformable/test/itkMultiResolutionPDEDeformableRegistrationTest.cxx
@@ -24,6 +24,7 @@
 
 #include <iostream>
 #include "itkTestingMacros.h"
+#include "itkArray.h"
 
 namespace
 {
@@ -243,18 +244,28 @@ itkMultiResolutionPDEDeformableRegistrationTest(int argc, char * argv[])
 
   constexpr unsigned int numLevel = 3;
   unsigned int           numIterations[numLevel];
+  itk::Array<unsigned int> numIterationsArray;
+
+  numIterationsArray.SetSize(numLevel);
   numIterations[0] = 64;
+  numIterationsArray[0] = numIterations[0];
 
   unsigned int ilevel;
   for (ilevel = 1; ilevel < numLevel; ++ilevel)
   {
     numIterations[ilevel] = numIterations[ilevel - 1] / 2;
+    numIterationsArray[ilevel] = numIterations[ilevel];
   }
 
   registrator->SetNumberOfLevels(numLevel);
   ITK_TEST_SET_GET_VALUE(numLevel, registrator->GetNumberOfLevels());
+  registrator->SetNumberOfLevels(numLevel + 2);
+  ITK_TEST_SET_GET_VALUE(numLevel + 2, registrator->GetNumberOfLevels());
+
+  registrator->SetNumberOfIterations(numIterationsArray);
+  // NumberOfLevels should now be numLevel because numIterationsArray is of size numLevel.
+  ITK_TEST_SET_GET_VALUE(numLevel, registrator->GetNumberOfLevels());
 
-  registrator->SetNumberOfIterations(numIterations);
   RegistrationType::NumberOfIterationsType numIterationsArr;
   numIterationsArr.SetData(numIterations, numLevel);
   ITK_TEST_SET_GET_VALUE(numIterationsArr, registrator->GetNumberOfIterations());