mirror of
https://github.com/OPM/ResInsight.git
synced 2026-10-26 07:00:17 -05:00
Magne Sjaastad
20a6373104
Add a test covering a flat 'realization-N/CASE.EXT' folder structure (no iter-N level) grouped with EnsembleGroupingMode::NONE, verifying all files end up in a single best-effort ensemble. Also add a test confirming empty input produces no bogus ensemble entry.
174 lines
8.6 KiB
C++
174 lines
8.6 KiB
C++
/////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (C) 2023 Equinor ASA
|
|
//
|
|
// ResInsight is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// ResInsight is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
// WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
// FITNESS FOR A PARTICULAR PURPOSE.
|
|
//
|
|
// See the GNU General Public License at <http://www.gnu.org/licenses/gpl.html>
|
|
// for more details.
|
|
//
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "RiaEnsembleNameTools.h"
|
|
|
|
#include <QStringList>
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
TEST( RiaEnsembleNameToolsTests, commonBaseName )
|
|
{
|
|
QStringList fileNames = { "/base/realization-0/iter-0/stimplan/output/B-H3_StimPlanModel_01_job-01/mainfrac/data_vs_time.csv",
|
|
"/base/realization-0/iter-0/stimplan/output/B-H3_StimPlanModel_02_job-02/mainfrac/data_vs_time.csv" };
|
|
|
|
ASSERT_EQ( "data_vs_time", RiaEnsembleNameTools::findCommonBaseName( fileNames ).toStdString() );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
TEST( RiaFilePathTools, EnsembleName )
|
|
{
|
|
{
|
|
QString testPath1( "e:/models/from_equinor_sftp/drogon3d_ahm/realization-0/iter-3/eclipse/model/DROGON-0.SMSPEC" );
|
|
QString testPath2( "e:/models/from_equinor_sftp/drogon3d_ahm/realization-1/iter-3/eclipse/model/DROGON-1.SMSPEC" );
|
|
QStringList allPaths = { testPath1, testPath2 };
|
|
|
|
auto ensembleName = RiaEnsembleNameTools::findSuitableEnsembleName( allPaths, RiaDefines::EnsembleGroupingMode::FMU_FOLDER_STRUCTURE );
|
|
|
|
EXPECT_EQ( QString( "iter-3" ), ensembleName );
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
TEST( RiaFilePathTools, RealizationName )
|
|
{
|
|
{
|
|
QString testPath0( "e:/models/from_equinor_sftp/drogon3d_ahm/realization-0/iter-3/eclipse/model/DROGON-0.SMSPEC" );
|
|
QString testPath1( "e:/models/from_equinor_sftp/drogon3d_ahm/realization-1/iter-3/eclipse/model/DROGON-1.SMSPEC" );
|
|
QStringList allPaths = { testPath0, testPath1 };
|
|
|
|
QString fileName = "DROGON-0.SMSPEC";
|
|
|
|
auto name = RiaEnsembleNameTools::uniqueShortName( testPath1, allPaths, fileName );
|
|
|
|
EXPECT_EQ( QString( "real-1" ), name );
|
|
}
|
|
|
|
{
|
|
QString testPath0( "/drogon3d_ahm/run-0/DROGON-0.SMSPEC" );
|
|
QString testPath1( "/drogon3d_ahm/run-1/DROGON-1.SMSPEC" );
|
|
QStringList allPaths = { testPath0, testPath1 };
|
|
|
|
QString fileName = "DROGON-0.SMSPEC";
|
|
|
|
auto name = RiaEnsembleNameTools::uniqueShortName( testPath1, allPaths, fileName );
|
|
|
|
EXPECT_EQ( QString( "run-1" ), name );
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
TEST( RiaFilePathTools, EnsembleGroupingMultipleEnsembles )
|
|
{
|
|
{
|
|
std::vector<std::string> filepaths = { "f:/Models/scratch/project_a/realization-0/iter-0/eclipse/model/PROJECT-0.SMSPEC",
|
|
"f:/Models/scratch/project_a/realization-1/iter-0/eclipse/model/PROJECT-1.SMSPEC",
|
|
"f:/Models/scratch/project_a/realization-22/iter-0/eclipse/model/PROJECT-22.SMSPEC",
|
|
"f:/Models/scratch/project_b/realization-0/iter-0/eclipse/model/PROJECT-0.SMSPEC",
|
|
"f:/Models/scratch/project_b/realization-1/iter-0/eclipse/model/PROJECT-1.SMSPEC",
|
|
"f:/Models/scratch/project_b/realization-2/iter-0/eclipse/model/PROJECT-2.SMSPEC",
|
|
"f:/Models/scratch/project_c/realization-0/iter-0/eclipse/model/PROJECT-0.SMSPEC",
|
|
"f:/Models/scratch/project_c/realization-1/iter-0/eclipse/model/PROJECT-1.SMSPEC",
|
|
"f:/Models/scratch/project_c/realization-2/iter-0/eclipse/model/PROJECT-2.SMSPEC",
|
|
"f:/Models/scratch/project_c/realization-0/iter-1/eclipse/model/PROJECT-0.SMSPEC",
|
|
"f:/Models/scratch/project_c/realization-1\\iter-1/eclipse/model\\PROJECT-1.SMSPEC",
|
|
"f:/Models/scratch/project_c/realization-\\iter-1/eclipse/model\\PROJECT-2.SMSPEC" };
|
|
|
|
auto grouping = RiaEnsembleNameTools::groupFilePathsFmu( filepaths );
|
|
|
|
std::set<std::string> keys1 = { "project_a", "project_b", "project_c" };
|
|
std::set<std::string> keys2 = { "iter-0", "iter-1" };
|
|
|
|
EXPECT_EQ( 4, grouping.size() );
|
|
for ( const auto& [keys, paths] : grouping )
|
|
{
|
|
const auto& [key1, key2] = keys;
|
|
EXPECT_TRUE( keys1.find( key1 ) != keys1.end() );
|
|
EXPECT_TRUE( keys2.find( key2 ) != keys2.end() );
|
|
}
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
// Some FMU-like data sets skip the "iter-N" folder level and place the case files directly inside
|
|
// "realization-N". Ensemble grouping mode NONE should still create a single best-effort ensemble
|
|
// containing all the files, regardless of the folder structure used.
|
|
//--------------------------------------------------------------------------------------------------
|
|
TEST( RiaFilePathTools, EnsembleGroupingNoneFlatRealizationStructure )
|
|
{
|
|
QStringList fileNames = { "f:/scratch/grid_ensemble/realization-1/SIMPLE_R1.EGRID",
|
|
"f:/scratch/grid_ensemble/realization-2/SIMPLE_R2.EGRID",
|
|
"f:/scratch/grid_ensemble/realization-3/SIMPLE_R3.EGRID" };
|
|
|
|
auto grouping = RiaEnsembleNameTools::groupFilesByEnsembleName( fileNames, RiaDefines::EnsembleGroupingMode::NONE );
|
|
|
|
ASSERT_EQ( 1u, grouping.size() );
|
|
|
|
const auto& [groupName, groupedFileNames] = *grouping.begin();
|
|
EXPECT_FALSE( groupName.isEmpty() );
|
|
EXPECT_EQ( 3, groupedFileNames.size() );
|
|
for ( const auto& fileName : fileNames )
|
|
{
|
|
EXPECT_TRUE( groupedFileNames.contains( fileName ) );
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
// groupFilesByEnsembleName with an empty input list must not create a bogus ensemble entry.
|
|
//--------------------------------------------------------------------------------------------------
|
|
TEST( RiaFilePathTools, EnsembleGroupingNoneEmptyInput )
|
|
{
|
|
QStringList fileNames;
|
|
|
|
auto grouping = RiaEnsembleNameTools::groupFilesByEnsembleName( fileNames, RiaDefines::EnsembleGroupingMode::NONE );
|
|
|
|
EXPECT_TRUE( grouping.empty() );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
TEST( RiaFilePathTools, EnsembleGroupingEverest )
|
|
{
|
|
{
|
|
std::vector<std::string> filepaths = {
|
|
"f:/Models/scratch/my_case/batch_0/geo_realization_0/simulation_0/eclipse/model/PROJECT-0.SMSPEC",
|
|
"f:/Models/scratch/my_case/batch_0/geo_realization_1/simulation_1/eclipse/model/PROJECT-0.SMSPEC",
|
|
"f:/Models/scratch/my_case/batch_0/geo_realization_2/simulation_2/eclipse/model/PROJECT-0.SMSPEC",
|
|
"f:/Models/scratch/my_case/batch_0/geo_realization_2/simulation_2/eclipse/model/PROJECT-0.SMSPEC",
|
|
"f:/Models/scratch/my_case/batch_1/geo_realization_0/simulation_0/eclipse/model/PROJECT-0.SMSPEC",
|
|
"f:/Models/scratch/my_case/batch_1/geo_realization_1/simulation_1/eclipse/model/PROJECT-0.SMSPEC",
|
|
"f:/Models/scratch/my_case/batch_1/geo_realization_2/simulation_2/eclipse/model/PROJECT-0.SMSPEC",
|
|
"f:/Models/scratch/my_case/batch_1/geo_realization_2/simulation_2/eclipse/model/PROJECT-0.SMSPEC",
|
|
};
|
|
|
|
auto grouping = RiaEnsembleNameTools::groupFilePathsEverest( filepaths );
|
|
|
|
std::set<std::string> keys1 = { "my_case" };
|
|
std::set<std::string> keys2 = { "batch_0", "batch_1" };
|
|
|
|
for ( const auto& [keys, paths] : grouping )
|
|
{
|
|
const auto& [key1, key2] = keys;
|
|
EXPECT_TRUE( keys1.find( key1 ) != keys1.end() );
|
|
EXPECT_TRUE( keys2.find( key2 ) != keys2.end() );
|
|
}
|
|
}
|
|
}
|