From b15502ddc861526a72f4041896fbabc5e0b64337 Mon Sep 17 00:00:00 2001 From: Richard Berger Date: Wed, 24 Mar 2021 15:53:00 -0400 Subject: [PATCH] Add utils::split_lines --- src/utils.cpp | 8 ++++++++ src/utils.h | 6 ++++++ unittest/utils/test_utils.cpp | 17 +++++++++++++---- 3 files changed, 27 insertions(+), 4 deletions(-) diff --git a/src/utils.cpp b/src/utils.cpp index b5576b9f27..e733d7eaae 100644 --- a/src/utils.cpp +++ b/src/utils.cpp @@ -851,6 +851,14 @@ std::vector utils::split_words(const std::string &text) return list; } +/* ---------------------------------------------------------------------- + Convert multi-line string into lines +------------------------------------------------------------------------- */ +std::vector utils::split_lines(const std::string &text) +{ + return Tokenizer(text, "\n").as_vector(); +} + /* ---------------------------------------------------------------------- Return whether string is a valid integer number ------------------------------------------------------------------------- */ diff --git a/src/utils.h b/src/utils.h index eab81f1343..ec4dd6ae85 100644 --- a/src/utils.h +++ b/src/utils.h @@ -321,6 +321,12 @@ namespace LAMMPS_NS { std::vector split_words(const std::string &text); + /** Take multi-line text and split into lines + * + * \param text string that should be split + * \return STL vector with the lines */ + std::vector split_lines(const std::string &text); + /** Check if string can be converted to valid integer * * \param str string that should be checked diff --git a/unittest/utils/test_utils.cpp b/unittest/utils/test_utils.cpp index 2fa17a5e5a..2a87f1c347 100644 --- a/unittest/utils/test_utils.cpp +++ b/unittest/utils/test_utils.cpp @@ -113,7 +113,7 @@ TEST(Utils, count_words_with_extra_spaces) TEST(Utils, split_words_simple) { - std::vector list = utils::split_words("one two three"); + auto list = utils::split_words("one two three"); ASSERT_EQ(list.size(), 3); ASSERT_THAT(list[0], StrEq("one")); ASSERT_THAT(list[1], StrEq("two")); @@ -122,7 +122,7 @@ TEST(Utils, split_words_simple) TEST(Utils, split_words_quoted) { - std::vector list = utils::split_words("one 'two' \"three\""); + auto list = utils::split_words("one 'two' \"three\""); ASSERT_EQ(list.size(), 3); ASSERT_THAT(list[0], StrEq("one")); ASSERT_THAT(list[1], StrEq("two")); @@ -131,7 +131,7 @@ TEST(Utils, split_words_quoted) TEST(Utils, split_words_escaped) { - std::vector list = utils::split_words("1\\' '\"two\"' 3\\\""); + auto list = utils::split_words("1\\' '\"two\"' 3\\\""); ASSERT_EQ(list.size(), 3); ASSERT_THAT(list[0], StrEq("1\\'")); ASSERT_THAT(list[1], StrEq("\"two\"")); @@ -140,13 +140,22 @@ TEST(Utils, split_words_escaped) TEST(Utils, split_words_quote_in_quoted) { - std::vector list = utils::split_words("one 't\\'wo' \"th\\\"ree\""); + auto list = utils::split_words("one 't\\'wo' \"th\\\"ree\""); ASSERT_EQ(list.size(), 3); ASSERT_THAT(list[0], StrEq("one")); ASSERT_THAT(list[1], StrEq("t\\'wo")); ASSERT_THAT(list[2], StrEq("th\\\"ree")); } +TEST(Utils, split_lines) +{ + auto list = utils::split_lines(" line 1\nline 2 \n line 3 \n"); + ASSERT_EQ(list.size(), 3); + ASSERT_THAT(list[0], StrEq(" line 1")); + ASSERT_THAT(list[1], StrEq("line 2 ")); + ASSERT_THAT(list[2], StrEq(" line 3 ")); +} + TEST(Utils, valid_integer1) { ASSERT_TRUE(utils::is_integer("10"));