Skip to content

Commit 54c8bf0

Browse files
committed
Change names with key_value to metadata
1 parent ee3df9e commit 54c8bf0

File tree

5 files changed

+25
-24
lines changed

5 files changed

+25
-24
lines changed

pineappl/src/fk_table.rs

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ use super::grid::Grid;
77
use super::pids::OptRules;
88
use super::subgrid::{self, EmptySubgridV1, Subgrid};
99
use ndarray::{s, ArrayD};
10+
use std::collections::BTreeMap;
1011
use std::fmt::{self, Display, Formatter};
1112
use std::iter;
1213
use std::str::FromStr;
@@ -215,11 +216,9 @@ impl FkTable {
215216
}
216217
}
217218

218-
/// Set a metadata key-value pair for this FK table.
219-
pub fn set_key_value(&mut self, key: &str, value: &str) {
220-
self.grid
221-
.metadata_mut()
222-
.insert(key.to_owned(), value.to_owned());
219+
/// Return the metadata of this FK-table.
220+
pub fn metadata_mut(&mut self) -> &mut BTreeMap<String, String> {
221+
self.grid.metadata_mut()
223222
}
224223

225224
/// Returns the x grid that all subgrids for all hadronic initial states share.

pineappl_py/src/fk_table.rs

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -248,8 +248,10 @@ impl PyFkTable {
248248
/// key
249249
/// value : str
250250
/// value
251-
pub fn set_key_value(&mut self, key: &str, value: &str) {
252-
self.fk_table.set_key_value(key, value);
251+
pub fn set_metadata(&mut self, key: &str, value: &str) {
252+
self.fk_table
253+
.metadata_mut()
254+
.insert(key.to_owned(), value.to_owned());
253255
}
254256

255257
/// Get metadata values stored in the grid.
@@ -261,7 +263,7 @@ impl PyFkTable {
261263
/// key, value map
262264
#[getter]
263265
#[must_use]
264-
pub fn key_values(&self) -> BTreeMap<String, String> {
266+
pub fn metadata(&self) -> BTreeMap<String, String> {
265267
self.fk_table.grid().metadata().clone()
266268
}
267269

pineappl_py/src/grid.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -332,7 +332,7 @@ impl PyGrid {
332332
/// key
333333
/// value : str
334334
/// value
335-
pub fn set_key_value(&mut self, key: &str, value: &str) {
335+
pub fn set_metadata(&mut self, key: &str, value: &str) {
336336
self.grid
337337
.metadata_mut()
338338
.insert(key.to_owned(), value.to_owned());
@@ -347,7 +347,7 @@ impl PyGrid {
347347
/// key, value map
348348
#[getter]
349349
#[must_use]
350-
pub fn key_values(&self) -> BTreeMap<String, String> {
350+
pub fn metadata(&self) -> BTreeMap<String, String> {
351351
self.grid.metadata().clone()
352352
}
353353

pineappl_py/tests/test_fk_table.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -88,13 +88,13 @@ def test_fktable(
8888
np.testing.assert_allclose(bin_limits[:, 0, 0], [0.0])
8989
np.testing.assert_allclose(bin_limits[:, 0, 1], [1.0])
9090

91-
# Check setting key values
92-
fk.set_key_value("bla", "blub")
93-
fk.set_key_value('"', "'")
94-
fk.set_key_value("äöü", \\")
95-
assert fk.key_values["bla"] == "blub"
96-
assert fk.key_values['"'] == "'"
97-
assert fk.key_values["äöü"] == \\"
91+
# Check setting metadata
92+
fk.set_metadata("bla", "blub")
93+
fk.set_metadata('"', "'")
94+
fk.set_metadata("äöü", \\")
95+
assert fk.metadata["bla"] == "blub"
96+
assert fk.metadata['"'] == "'"
97+
assert fk.metadata["äöü"] == \\"
9898

9999
# Check the various aspects of the Channels
100100
channels = fk.channels()

pineappl_py/tests/test_grid.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -609,20 +609,20 @@ def test_io(self, tmp_path, fake_grids):
609609
assert isinstance(gg, Grid)
610610
_ = Grid.read(str(p))
611611

612-
def test_key_values(self, fake_grids):
612+
def test_metadata(self, fake_grids):
613613
g = fake_grids.grid_with_generic_convolution(
614614
nb_convolutions=2,
615615
channels=CHANNELS,
616616
orders=ORDERS,
617617
convolutions=[CONVOBJECT, CONVOBJECT],
618618
)
619-
g.set_key_value("bla", "blub")
620-
g.set_key_value('"', "'")
621-
g.set_key_value("äöü", \\")
619+
g.set_metadata("bla", "blub")
620+
g.set_metadata('"', "'")
621+
g.set_metadata("äöü", \\")
622622

623-
assert g.key_values["bla"] == "blub"
624-
assert g.key_values['"'] == "'"
625-
assert g.key_values["äöü"] == \\"
623+
assert g.metadata["bla"] == "blub"
624+
assert g.metadata['"'] == "'"
625+
assert g.metadata["äöü"] == \\"
626626

627627
def test_pid_basis(self, fake_grids):
628628
g = fake_grids.grid_with_generic_convolution(

0 commit comments

Comments
 (0)