diff --git a/opm/parser/eclipse/Deck/DeckIntItem.cpp b/opm/parser/eclipse/Deck/DeckIntItem.cpp index 069a027a7..6acb2ad17 100644 --- a/opm/parser/eclipse/Deck/DeckIntItem.cpp +++ b/opm/parser/eclipse/Deck/DeckIntItem.cpp @@ -34,5 +34,14 @@ namespace Opm { } } + void DeckIntItem::push_back(int data) { + m_data.push_back( data ); + } + + + size_t DeckIntItem::size() const { + return m_data.size(); + } + } diff --git a/opm/parser/eclipse/Deck/DeckIntItem.hpp b/opm/parser/eclipse/Deck/DeckIntItem.hpp index e39f0d64d..4416c0f8a 100644 --- a/opm/parser/eclipse/Deck/DeckIntItem.hpp +++ b/opm/parser/eclipse/Deck/DeckIntItem.hpp @@ -18,7 +18,7 @@ */ #ifndef DECKINTITEM_HPP -#define DECKINTITEM_HPP +#define DECKINTITEM_HPP #include #include @@ -30,6 +30,8 @@ namespace Opm { public: int getInt(unsigned int index) const; void push_back(std::vector data); + void push_back(int value); + size_t size() const; private: std::vector m_data; }; @@ -37,5 +39,5 @@ namespace Opm { typedef boost::shared_ptr DeckIntItemPtr; typedef boost::shared_ptr DeckIntItemConstPtr; } -#endif /* DECKINTITEM_HPP */ +#endif /* DECKINTITEM_HPP */ diff --git a/opm/parser/eclipse/Deck/tests/DeckItemTests.cpp b/opm/parser/eclipse/Deck/tests/DeckItemTests.cpp index 1526181f3..14df0ea05 100644 --- a/opm/parser/eclipse/Deck/tests/DeckItemTests.cpp +++ b/opm/parser/eclipse/Deck/tests/DeckItemTests.cpp @@ -44,3 +44,17 @@ BOOST_AUTO_TEST_CASE(PushBack_VectorPushed_ElementsCorrect) { BOOST_CHECK_EQUAL(13, deckIntItem.getInt(0)); BOOST_CHECK_EQUAL(33, deckIntItem.getInt(1)); } + + + +BOOST_AUTO_TEST_CASE(size_correct) { + DeckIntItem deckIntItem; + + BOOST_CHECK_EQUAL( 0U , deckIntItem.size()); + deckIntItem.push_back( 100 ); + BOOST_CHECK_EQUAL( 1U , deckIntItem.size()); + + deckIntItem.push_back( 100 ); + deckIntItem.push_back( 100 ); + BOOST_CHECK_EQUAL( 3U , deckIntItem.size()); +}