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

Add more tests for COMPRESS package

Browse Source
This commit is contained in:
Richard Berger
2021-03-11 14:21:16 -05:00
parent 14da94d189
commit 76d857e428
12 changed files with 129 additions and 219 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
unittest/formats/test_dump_atom_compressed.cpp
View File

@ -40,7 +40,11 @@ TEST_F(DumpAtomCompressTest, compressed_run0)
auto text_file = text_dump_filename("run0.melt");
auto compressed_file = compressed_dump_filename("run0.melt");
generate_text_and_compressed_dump(text_file, compressed_file, "", "", 0);
if(compression_style == "atom/zstd") {
generate_text_and_compressed_dump(text_file, compressed_file, "", "", "", "checksum yes", 0);
} else {
generate_text_and_compressed_dump(text_file, compressed_file, "", "", 0);
}
TearDown();
@ -71,7 +75,11 @@ TEST_F(DumpAtomCompressTest, compressed_multi_file_run1)
auto compressed_file_0 = compressed_dump_filename(base_name_0);
auto compressed_file_1 = compressed_dump_filename(base_name_1);
generate_text_and_compressed_dump(text_file, compressed_file, "", "", 1);
if(compression_style == "atom/zstd") {
generate_text_and_compressed_dump(text_file, compressed_file, "", "", "", "checksum no", 1);
} else {
generate_text_and_compressed_dump(text_file, compressed_file, "", "", 1);
}
TearDown();
@ -378,43 +386,3 @@ TEST_F(DumpAtomCompressTest, compressed_modify_clevel_run0)
delete_file(compressed_file);
delete_file(converted_file);
}
int main(int argc, char **argv)
{
MPI_Init(&argc, &argv);
::testing::InitGoogleMock(&argc, argv);
if (argc < 2) {
std::cerr << "usage: " << argv[0] << " (gz|zstd)\n\n" << std::endl;
return 1;
}
if(strcmp(argv[1], "gz") == 0) {
COMPRESS_SUFFIX = "gz";
COMPRESS_EXTENSION = "gz";
} else if(strcmp(argv[1], "zstd") == 0) {
COMPRESS_SUFFIX = "zstd";
COMPRESS_EXTENSION = "zst";
} else {
std::cerr << "usage: " << argv[0] << " (gz|zstd)\n\n" << std::endl;
return 1;
}
COMPRESS_BINARY = getenv("COMPRESS_BINARY");
// handle arguments passed via environment variable
if (const char *var = getenv("TEST_ARGS")) {
std::vector<std::string> env = utils::split_words(var);
for (auto arg : env) {
if (arg == "-v") {
verbose = true;
}
}
}
if ((argc > 1) && (strcmp(argv[1], "-v") == 0)) verbose = true;
int rv = RUN_ALL_TESTS();
MPI_Finalize();
return rv;
}