Skip to content
Snippets Groups Projects
Commit 745f8f6a authored by Jordi Inglada's avatar Jordi Inglada
Browse files

ADD: test for shark utils

parent 75663d12
No related branches found
No related tags found
No related merge requests found
/*
* Copyright (C) 2005-2017 Centre National d'Etudes Spatiales (CNES)
*
* This file is part of Orfeo Toolbox
*
* https://www.orfeo-toolbox.org/
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "itkMacro.h"
#include "otbSharkUtils.h"
int otbSharkNormalizeLabels(int itkNotUsed(argc), char* itkNotUsed(argv) [])
{
std::vector<unsigned int> inLabels = {2, 2, 3, 20, 1};
std::vector<unsigned int> expectedDictionary = {2, 3, 20, 1};
std::vector<unsigned int> expectedLabels = {0, 0, 1, 2, 3};
auto newLabels = inLabels;
std::vector<unsigned int> labelDict;
otb::Shark::NormalizeLabelsAndGetDictionary(newLabels, labelDict);
if(newLabels != expectedLabels)
{
std::cout << "Wrong new labels\n";
for(size_t i = 0; i<newLabels.size(); ++i)
std::cout << "Got " << newLabels[i] << " expected " << expectedLabels[i] << '\n';
return EXIT_FAILURE;
}
if(labelDict != expectedDictionary)
{
std::cout << "Wrong dictionary\n";
for(size_t i = 0; i<labelDict.size(); ++i)
std::cout << "Got " << labelDict[i] << " expected " << expectedDictionary[i] << '\n';
return EXIT_FAILURE;
}
return EXIT_SUCCESS;
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment