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")