From 576ddd49876d4d71d003eab15b78941d99deb25f Mon Sep 17 00:00:00 2001 From: Steinar Foss Date: Fri, 25 Oct 2019 14:11:52 +0200 Subject: [PATCH] python deck: added member add(keyword). --- python/cxx/deck.cpp | 6 ++++++ python/tests/test_parser.py | 4 ++++ 2 files changed, 10 insertions(+) diff --git a/python/cxx/deck.cpp b/python/cxx/deck.cpp index d6024fc16..2014890c7 100644 --- a/python/cxx/deck.cpp +++ b/python/cxx/deck.cpp @@ -35,6 +35,11 @@ namespace { return deck.getKeyword(index); } + //This adds a keyword by copy + void addKeyword(Deck& deck, const DeckKeyword kw) { + deck.addKeyword(kw); + } + } @@ -59,6 +64,7 @@ void python::common::export_Deck(py::module &module) { } ) .def( "count", &count ) + .def( "add", &addKeyword) ; } diff --git a/python/tests/test_parser.py b/python/tests/test_parser.py index f00283d53..eff64cb49 100644 --- a/python/tests/test_parser.py +++ b/python/tests/test_parser.py @@ -147,6 +147,10 @@ FIPNUM si_array = zcorn_kw.get_SI_array() self.assertAlmostEqual( si_array[0], 1.1 * unit_foot ) self.assertAlmostEqual( si_array[2], 3.3 * unit_foot ) + + assert( not( "ZCORN" in deck ) ) + deck.add( zcorn_kw ) + assert( "ZCORN" in deck )