def test_create_one_region():
cfg = {"_" + k: v for k, v in binning.bins_nmuon.items()}
_in, _out, _bins, _index = mgr.create_one_dimension("test_create_one_region", **cfg)
assert _in == "NMuon"
assert _out == "nmuon"
assert _index is None
assert _bins is None
def test_create_one_dimension_HT():
cfg = {"_" + k: v for k, v in binning.bins_py.items()}
_in, _out, _bins, _index = mgr.create_one_dimension("test_create_one_dimension_HT", **cfg)
assert _in == "Jet_Py"
assert _out == "py_leadJet"
assert _index == 0
assert isinstance(_bins, np.ndarray)
assert np.all(_bins[1:-1] == [0, 20, 100])
assert _bins[0] == float("-inf")
assert _bins[-1] == float("inf")
def test_create_one_dimension_aT():
cfg = {"_" + k: v for k, v in binning.bins_met_px.items()}
_in, _out, _bins, _index = mgr.create_one_dimension("test_create_one_dimension_aT", **cfg)
assert _in == "MET_px"
assert _out == "met_px"
assert _index is None
assert isinstance(_bins, np.ndarray)
assert np.all(_bins[1:-1] == np.linspace(0, 100, 11))
assert _bins[0] == float("-inf")
assert _bins[-1] == float("inf")
def test_create_one_dimension_aT():
cfg = {"_" + k: v for k, v in binning.bins_met_px.items()}
_in, _out, _bins, _index = mgr.create_one_dimension(
"test_create_one_dimension_aT", **cfg)
assert _in == "MET_px"
assert _out == "met_px"
assert _index is None
assert isinstance(_bins, pd.IntervalIndex)
assert np.all(_bins.left[1:] == np.linspace(0, 100, 30))
assert _bins[0].left == float("-inf")
assert _bins[-1].right == float("inf")