ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add more tests for dump atom/gz

Browse Source
This commit is contained in:
Richard Berger
2021-03-10 13:40:10 -05:00
parent 6b3f0ddd3b
commit a548ea3bca
2 changed files with 177 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

177
unittest/formats/test_dump_atom_gz.cpp
View File

@ -52,14 +52,25 @@ public:
void generate_text_and_compressed_dump(std::string text_file, std::string compressed_file, void generate_text_and_compressed_dump(std::string text_file, std::string compressed_file,
std::string compression_style, std::string compression_style,
std::string dump_modify_options, int ntimesteps) std::string dump_modify_options, int ntimesteps)
{
generate_text_and_compressed_dump(text_file, compressed_file, compression_style,
dump_modify_options, dump_modify_options, ntimesteps);
}
void generate_text_and_compressed_dump(std::string text_file, std::string compressed_file,
std::string compression_style,
std::string text_options, std::string compressed_options, int ntimesteps)
{ {
if (!verbose) ::testing::internal::CaptureStdout(); if (!verbose) ::testing::internal::CaptureStdout();
command(fmt::format("dump id0 all {} 1 {}", dump_style, text_file)); command(fmt::format("dump id0 all {} 1 {}", dump_style, text_file));
command(fmt::format("dump id1 all {} 1 {}", compression_style, compressed_file)); command(fmt::format("dump id1 all {} 1 {}", compression_style, compressed_file));
if (!dump_modify_options.empty()) { if (!text_options.empty()) {
command(fmt::format("dump_modify id0 {}", dump_modify_options)); command(fmt::format("dump_modify id0 {}", text_options));
command(fmt::format("dump_modify id1 {}", dump_modify_options)); }
if (!compressed_options.empty()) {
command(fmt::format("dump_modify id1 {}", compressed_options));
} }
command(fmt::format("run {}", ntimesteps)); command(fmt::format("run {}", ntimesteps));
@ -106,6 +117,124 @@ TEST_F(DumpAtomGZTest, compressed_run0)
delete_file(converted_file); delete_file(converted_file);
} }
TEST_F(DumpAtomGZTest, compressed_multi_file_run1)
{
if (!GZIP_BINARY) GTEST_SKIP();
auto text_file = "dump_gz_text_multi_file_run1_*.melt";
auto compressed_file = "dump_gz_compressed_multi_file_run1_*.melt.gz";
auto text_file_0 = "dump_gz_text_multi_file_run1_0.melt";
auto text_file_1 = "dump_gz_text_multi_file_run1_1.melt";
auto compressed_file_0 = "dump_gz_compressed_multi_file_run1_0.melt.gz";
auto compressed_file_1 = "dump_gz_compressed_multi_file_run1_1.melt.gz";
generate_text_and_compressed_dump(text_file, compressed_file, "atom/gz", "", 1);
TearDown();
ASSERT_FILE_EXISTS(text_file_0);
ASSERT_FILE_EXISTS(text_file_1);
ASSERT_FILE_EXISTS(compressed_file_0);
ASSERT_FILE_EXISTS(compressed_file_1);
auto converted_file_0 = convert_compressed_to_text(compressed_file_0);
auto converted_file_1 = convert_compressed_to_text(compressed_file_1);
ASSERT_THAT(converted_file_0, Eq("dump_gz_compressed_multi_file_run1_0.melt"));
ASSERT_THAT(converted_file_1, Eq("dump_gz_compressed_multi_file_run1_1.melt"));
ASSERT_FILE_EXISTS(converted_file_0);
ASSERT_FILE_EXISTS(converted_file_1);
ASSERT_FILE_EQUAL(text_file_0, converted_file_0);
ASSERT_FILE_EQUAL(text_file_1, converted_file_1);
delete_file(text_file_0);
delete_file(text_file_1);
delete_file(compressed_file_0);
delete_file(compressed_file_1);
delete_file(converted_file_0);
delete_file(converted_file_1);
}
TEST_F(DumpAtomGZTest, compressed_multi_file_with_pad_run1)
{
if (!GZIP_BINARY) GTEST_SKIP();
auto text_file = "dump_gz_text_multi_file_pad_run1_*.melt";
auto compressed_file = "dump_gz_compressed_multi_file_pad_run1_*.melt.gz";
auto text_file_0 = "dump_gz_text_multi_file_pad_run1_000.melt";
auto text_file_1 = "dump_gz_text_multi_file_pad_run1_001.melt";
auto compressed_file_0 = "dump_gz_compressed_multi_file_pad_run1_000.melt.gz";
auto compressed_file_1 = "dump_gz_compressed_multi_file_pad_run1_001.melt.gz";
generate_text_and_compressed_dump(text_file, compressed_file, "atom/gz", "pad 3", 1);
TearDown();
ASSERT_FILE_EXISTS(text_file_0);
ASSERT_FILE_EXISTS(text_file_1);
ASSERT_FILE_EXISTS(compressed_file_0);
ASSERT_FILE_EXISTS(compressed_file_1);
auto converted_file_0 = convert_compressed_to_text(compressed_file_0);
auto converted_file_1 = convert_compressed_to_text(compressed_file_1);
ASSERT_THAT(converted_file_0, Eq("dump_gz_compressed_multi_file_pad_run1_000.melt"));
ASSERT_THAT(converted_file_1, Eq("dump_gz_compressed_multi_file_pad_run1_001.melt"));
ASSERT_FILE_EXISTS(converted_file_0);
ASSERT_FILE_EXISTS(converted_file_1);
ASSERT_FILE_EQUAL(text_file_0, converted_file_0);
ASSERT_FILE_EQUAL(text_file_1, converted_file_1);
delete_file(text_file_0);
delete_file(text_file_1);
delete_file(compressed_file_0);
delete_file(compressed_file_1);
delete_file(converted_file_0);
delete_file(converted_file_1);
}
TEST_F(DumpAtomGZTest, compressed_multi_file_with_maxfiles_run1)
{
if (!GZIP_BINARY) GTEST_SKIP();
auto text_file = "dump_gz_text_multi_file_run1_*.melt";
auto compressed_file = "dump_gz_compressed_multi_file_run1_*.melt.gz";
auto text_file_0 = "dump_gz_text_multi_file_run1_0.melt";
auto text_file_1 = "dump_gz_text_multi_file_run1_1.melt";
auto text_file_2 = "dump_gz_text_multi_file_run1_2.melt";
auto compressed_file_0 = "dump_gz_compressed_multi_file_run1_0.melt.gz";
auto compressed_file_1 = "dump_gz_compressed_multi_file_run1_1.melt.gz";
auto compressed_file_2 = "dump_gz_compressed_multi_file_run1_2.melt.gz";
generate_text_and_compressed_dump(text_file, compressed_file, "atom/gz", "maxfiles 2", 2);
TearDown();
ASSERT_FILE_NOT_EXISTS(text_file_0);
ASSERT_FILE_EXISTS(text_file_1);
ASSERT_FILE_EXISTS(text_file_2);
ASSERT_FILE_NOT_EXISTS(compressed_file_0);
ASSERT_FILE_EXISTS(compressed_file_1);
ASSERT_FILE_EXISTS(compressed_file_2);
auto converted_file_1 = convert_compressed_to_text(compressed_file_1);
auto converted_file_2 = convert_compressed_to_text(compressed_file_2);
ASSERT_THAT(converted_file_1, Eq("dump_gz_compressed_multi_file_run1_1.melt"));
ASSERT_THAT(converted_file_2, Eq("dump_gz_compressed_multi_file_run1_2.melt"));
ASSERT_FILE_EXISTS(converted_file_1);
ASSERT_FILE_EXISTS(converted_file_2);
ASSERT_FILE_EQUAL(text_file_1, converted_file_1);
ASSERT_FILE_EQUAL(text_file_2, converted_file_2);
delete_file(text_file_1);
delete_file(text_file_2);
delete_file(compressed_file_1);
delete_file(compressed_file_2);
delete_file(converted_file_1);
delete_file(converted_file_2);
}
TEST_F(DumpAtomGZTest, compressed_with_units_run0) TEST_F(DumpAtomGZTest, compressed_with_units_run0)
{ {
if (!GZIP_BINARY) GTEST_SKIP(); if (!GZIP_BINARY) GTEST_SKIP();
@ -252,6 +381,48 @@ TEST_F(DumpAtomGZTest, compressed_triclinic_with_image_run0)
delete_file(converted_file); delete_file(converted_file);
} }
TEST_F(DumpAtomGZTest, compressed_modify_bad_param)
{
command("dump id1 all atom/gz 1 dump_gz_text_modify_bad_param_run0_*.melt.gz");
TEST_FAILURE(".*ERROR: Illegal dump_modify command: compression level must in the range of.*",
command("dump_modify id1 compression_level 12");
);
}
TEST_F(DumpAtomGZTest, compressed_modify_multi_bad_param)
{
command("dump id1 all atom/gz 1 dump_gz_text_modify_bad_param_run0_*.melt.gz");
TEST_FAILURE(".*ERROR: Illegal dump_modify command: compression level must in the range of.*",
command("dump_modify id1 pad 3 compression_level 12");
);
}
TEST_F(DumpAtomGZTest, compressed_modify_clevel_run0)
{
if (!GZIP_BINARY) GTEST_SKIP();
auto text_file = "dump_gz_text_modify_clevel_run0.melt";
auto compressed_file = "dump_gz_compressed_modify_clevel_run0.melt.gz";
generate_text_and_compressed_dump(text_file, compressed_file, "atom/gz", "", "compression_level 3", 0);
TearDown();
ASSERT_FILE_EXISTS(text_file);
ASSERT_FILE_EXISTS(compressed_file);
auto converted_file = convert_compressed_to_text(compressed_file);
ASSERT_THAT(converted_file, Eq("dump_gz_compressed_modify_clevel_run0.melt"));
ASSERT_FILE_EXISTS(converted_file);
ASSERT_FILE_EQUAL(text_file, converted_file);
delete_file(text_file);
delete_file(compressed_file);
delete_file(converted_file);
}
int main(int argc, char **argv) int main(int argc, char **argv)
{ {
MPI_Init(&argc, &argv); MPI_Init(&argc, &argv);

3
unittest/testing/utils.h
View File

@ -70,6 +70,9 @@ static bool file_exists(const std::string &filename)
} }
#define ASSERT_FILE_EXISTS(NAME) ASSERT_TRUE(file_exists(NAME)) #define ASSERT_FILE_EXISTS(NAME) ASSERT_TRUE(file_exists(NAME))
#define ASSERT_FILE_NOT_EXISTS(NAME) ASSERT_FALSE(file_exists(NAME))
#define ASSERT_FILE_EQUAL(FILE_A, FILE_B) ASSERT_TRUE(equal_lines(FILE_A, FILE_B)) #define ASSERT_FILE_EQUAL(FILE_A, FILE_B) ASSERT_TRUE(equal_lines(FILE_A, FILE_B))
#endif #endif