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

add unit tests for lammps_extract_variable()

Browse Source
This commit is contained in:
Axel Kohlmeyer
2024-06-30 18:32:37 -04:00
parent f8a7fc787c
commit d75e66dc71
3 changed files with 222 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
unittest/c-library/CMakeLists.txt
View File

@ -18,6 +18,12 @@ target_compile_definitions(test_library_properties PRIVATE -DTEST_INPUT_FOLDER=$
add_test(NAME LibraryProperties COMMAND test_library_properties) add_test(NAME LibraryProperties COMMAND test_library_properties)
set_tests_properties(LibraryProperties PROPERTIES ENVIRONMENT "LAMMPS_POTENTIALS=${LAMMPS_POTENTIALS_DIR}") set_tests_properties(LibraryProperties PROPERTIES ENVIRONMENT "LAMMPS_POTENTIALS=${LAMMPS_POTENTIALS_DIR}")
add_executable(test_library_objects test_library_objects.cpp test_main.cpp)
target_link_libraries(test_library_objects PRIVATE lammps GTest::GMock)
target_compile_definitions(test_library_objects PRIVATE -DTEST_INPUT_FOLDER=${CMAKE_CURRENT_SOURCE_DIR})
add_test(NAME LibraryObjects COMMAND test_library_objects)
set_tests_properties(LibraryObjects PROPERTIES ENVIRONMENT "LAMMPS_POTENTIALS=${LAMMPS_POTENTIALS_DIR}")
add_executable(test_library_scatter_gather test_library_scatter_gather.cpp test_main.cpp) add_executable(test_library_scatter_gather test_library_scatter_gather.cpp test_main.cpp)
target_link_libraries(test_library_scatter_gather PRIVATE lammps GTest::GMock) target_link_libraries(test_library_scatter_gather PRIVATE lammps GTest::GMock)
target_compile_definitions(test_library_scatter_gather PRIVATE -DTEST_INPUT_FOLDER=${CMAKE_CURRENT_SOURCE_DIR}) target_compile_definitions(test_library_scatter_gather PRIVATE -DTEST_INPUT_FOLDER=${CMAKE_CURRENT_SOURCE_DIR})

215
unittest/c-library/test_library_objects.cpp Normal file
View File

@ -0,0 +1,215 @@
// unit tests for checking and changing simulation properties through the library interface
#include "library.h"
#include "atom.h"
#include "input.h"
#include "lammps.h"
#include "lmptype.h"
#include "platform.h"
#include "variable.h"
#include <string>
#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include "test_main.h"
#define STRINGIFY(val) XSTR(val)
#define XSTR(val) #val
using ::LAMMPS_NS::Atom;
using ::LAMMPS_NS::bigint;
using ::LAMMPS_NS::Input;
using ::LAMMPS_NS::tagint;
using ::LAMMPS_NS::Variable;
using ::testing::HasSubstr;
using ::testing::StartsWith;
using ::testing::StrEq;
class LibraryObjects : public ::testing::Test {
protected:
void *lmp;
Variable *variable;
std::string INPUT_DIR = STRINGIFY(TEST_INPUT_FOLDER);
LibraryObjects() = default;
~LibraryObjects() override = default;
void SetUp() override
{
const char *args[] = {"LAMMPS_test", "-log", "none",
"-echo", "screen", "-nocite",
"-var", "input_dir", STRINGIFY(TEST_INPUT_FOLDER),
nullptr};
char **argv = (char **)args;
int argc = (sizeof(args) / sizeof(char *)) - 1;
::testing::internal::CaptureStdout();
lmp = lammps_open_no_mpi(argc, argv, nullptr);
variable = ((LAMMPS_NS::LAMMPS *)lmp)->input->variable;
std::string output = ::testing::internal::GetCapturedStdout();
if (verbose) std::cout << output;
EXPECT_THAT(output, StartsWith("LAMMPS ("));
}
void TearDown() override
{
::testing::internal::CaptureStdout();
lammps_close(lmp);
std::string output = ::testing::internal::GetCapturedStdout();
EXPECT_THAT(output, HasSubstr("Total wall time:"));
if (verbose) std::cout << output;
lmp = nullptr;
}
};
TEST_F(LibraryObjects, variables)
{
FILE *fp = fopen("test_variable.file", "w");
fputs("# test file for file style variable\n\n\none\n two \n\n"
"three # with comment\nfour ! with non-comment\n"
"# comments only\n five\n#END\n",
fp);
fclose(fp);
::testing::internal::CaptureStdout();
lammps_command(lmp, "region box block 0 2 0 2 0 2");
lammps_command(lmp, "create_box 1 box");
lammps_command(lmp, "mass 1 3.0");
lammps_command(lmp, "create_atoms 1 single 1.0 1.0 1.5");
lammps_command(lmp, "create_atoms 1 single 0.2 0.1 0.1");
lammps_command(lmp, "shell putenv TEST_VARIABLE=simpletest2");
lammps_command(lmp, "shell putenv TEST_VARIABLE2=simpletest OTHER_VARIABLE=2");
lammps_command(lmp, "variable one index 1 2 3 4");
lammps_command(lmp, "variable two equal 1");
lammps_command(lmp, "variable two equal 2");
lammps_command(lmp, "variable three string four");
lammps_command(lmp, "variable three string three");
lammps_command(lmp, "variable four1 loop 4");
lammps_command(lmp, "variable four2 loop 2 4");
lammps_command(lmp, "variable five1 loop 100 pad");
lammps_command(lmp, "variable five2 loop 10 200 pad");
lammps_command(lmp, "variable six world one");
lammps_command(lmp, "variable seven format two \"%5.2f\"");
lammps_command(lmp, "variable eight getenv TEST_VARIABLE2");
lammps_command(lmp, "variable eight getenv XXX");
lammps_command(lmp, "variable nine file test_variable.file");
lammps_command(lmp, "variable ten internal 1.0");
lammps_command(lmp, "variable ten internal 10.0");
lammps_command(lmp, "variable ten1 universe 1 2 3 4");
lammps_command(lmp, "variable ten2 uloop 4");
lammps_command(lmp, "variable ten3 uloop 4 pad");
lammps_command(lmp, "variable ten4 vector [0,1,2,3,5,7,11]");
lammps_command(lmp, "variable ten5 vector [0.5,1.25]");
lammps_command(lmp, "variable dummy index 0");
lammps_command(lmp, "variable file equal is_file(MYFILE)");
lammps_command(lmp, "variable iswin equal is_os(^Windows)");
lammps_command(lmp, "variable islin equal is_os(^Linux)");
std::string output = ::testing::internal::GetCapturedStdout();
if (verbose) std::cout << output;
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "unknown"), -1);
void *ptr = lammps_extract_variable(lmp, "unknown", NULL);
EXPECT_EQ(ptr, nullptr);
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "one"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "one", NULL);
EXPECT_NE(ptr, nullptr);
EXPECT_THAT((char *)ptr, StrEq("1"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "two"), LMP_VAR_EQUAL);
ptr = lammps_extract_variable(lmp, "two", NULL);
EXPECT_NE(ptr, nullptr);
EXPECT_THAT(*(double *)ptr, 2.0);
lammps_free(ptr);
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "three"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "three", NULL);
EXPECT_THAT((char *)ptr, StrEq("three"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "four1"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "four1", NULL);
EXPECT_THAT((char *)ptr, StrEq("1"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "four2"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "four2", NULL);
EXPECT_THAT((char *)ptr, StrEq("2"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "five1"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "five1", NULL);
EXPECT_THAT((char *)ptr, StrEq("001"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "five2"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "five2", NULL);
EXPECT_THAT((char *)ptr, StrEq("010"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "six"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "six", NULL);
EXPECT_THAT((char *)ptr, StrEq("one"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "seven"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "seven", NULL);
EXPECT_THAT((char *)ptr, StrEq(" 2.00"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "eight"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "eight", NULL);
EXPECT_THAT((char *)ptr, StrEq(""));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "nine"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "nine", NULL);
EXPECT_THAT((char *)ptr, StrEq("one"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "ten"), LMP_VAR_EQUAL);
ptr = lammps_extract_variable(lmp, "ten", NULL);
EXPECT_THAT(*(double *)ptr, 1.0);
lammps_free(ptr);
variable->internal_set(variable->find("ten"), 2.5);
ptr = lammps_extract_variable(lmp, "ten", NULL);
EXPECT_THAT(*(double *)ptr, 2.5);
lammps_free(ptr);
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "ten1"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "ten1", NULL);
EXPECT_THAT((char *)ptr, StrEq("1"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "ten2"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "ten2", NULL);
EXPECT_THAT((char *)ptr, StrEq("1"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "ten3"), LMP_VAR_STRING);
ptr = lammps_extract_variable(lmp, "ten3", NULL);
EXPECT_THAT((char *)ptr, StrEq("1"));
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "ten4"), LMP_VAR_VECTOR);
ptr = lammps_extract_variable(lmp, "ten4", (const char *)1);
double *dptr = (double *)lammps_extract_variable(lmp, "ten4", NULL);
EXPECT_EQ((*(int *)ptr), 7);
lammps_free(ptr);
EXPECT_EQ(dptr[0], 0);
EXPECT_EQ(dptr[4], 5);
EXPECT_EQ(dptr[6], 11);
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "ten5"), LMP_VAR_VECTOR);
ptr = lammps_extract_variable(lmp, "ten5", (const char *)1);
dptr = (double *)lammps_extract_variable(lmp, "ten5", NULL);
EXPECT_EQ((*(int *)ptr), 2);
lammps_free(ptr);
EXPECT_EQ(dptr[0], 0.5);
EXPECT_EQ(dptr[1], 1.25);
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "iswin"), LMP_VAR_EQUAL);
EXPECT_EQ(lammps_extract_variable_datatype(lmp, "islin"), LMP_VAR_EQUAL);
#if defined(_WIN32)
ptr = lammps_extract_variable(lmp, "iswin", NULL);
EXPECT_THAT(*(double *)ptr, 1.0);
lammps_free(ptr);
ptr = lammps_extract_variable(lmp, "islin", NULL);
EXPECT_THAT(*(double *)ptr, 0.0);
lammps_free(ptr);
#elif defined(__linux__)
ptr = lammps_extract_variable(lmp, "iswin", NULL);
EXPECT_THAT(*(double *)ptr, 0.0);
lammps_free(ptr);
ptr = lammps_extract_variable(lmp, "islin", NULL);
EXPECT_THAT(*(double *)ptr, 1.0);
lammps_free(ptr);
#else
ptr = lammps_extract_variable(lmp, "iswin", NULL);
EXPECT_THAT(*(double *)ptr, 0.0);
lammps_free(ptr);
ptr = lammps_extract_variable(lmp, "islin", NULL);
EXPECT_THAT(*(double *)ptr, 0.0);
lammps_free(ptr);
#endif
LAMMPS_NS::platform::unlink("test_variable.file");
}

1
unittest/commands/test_variables.cpp
View File

@ -157,6 +157,7 @@ TEST_F(VariableTest, CreateDelete)
ASSERT_THAT(variable->retrieve("three"), StrEq("four")); ASSERT_THAT(variable->retrieve("three"), StrEq("four"));
ASSERT_THAT(variable->retrieve("four2"), StrEq("2")); ASSERT_THAT(variable->retrieve("four2"), StrEq("2"));
ASSERT_THAT(variable->retrieve("five1"), StrEq("001")); ASSERT_THAT(variable->retrieve("five1"), StrEq("001"));
ASSERT_THAT(variable->retrieve("five2"), StrEq("010"));
ASSERT_THAT(variable->retrieve("seven"), StrEq(" 2.00")); ASSERT_THAT(variable->retrieve("seven"), StrEq(" 2.00"));
ASSERT_THAT(variable->retrieve("ten"), StrEq("1")); ASSERT_THAT(variable->retrieve("ten"), StrEq("1"));
ASSERT_THAT(variable->retrieve("eight"), StrEq("")); ASSERT_THAT(variable->retrieve("eight"), StrEq(""));