class TestIdMapperByCode(unittest.TestCase):
def setUp(self):
h5py_file = h5py.File(grid_file, "r")
self.id_mapper = IdMapper(H5pyGroup(h5py_file, "mappings")["id_map"])
def test_get_channels_by_code(self):
tc = TYPE_CODE_MAP["v2_channel"]
qs = self.id_mapper.get_by_code(tc)
assert np.all(qs["obj_code"] == tc)
def test_get_culvert_by_code(self):
tc = TYPE_CODE_MAP["v2_culvert"]
qs = self.id_mapper.get_by_code(tc)
assert np.all(qs["obj_code"] == tc)
def test_init_idMapper(mocked_id_map, h5py_file, threedi_datasource):
mocked_id_map.return_value = simple_id_map()
IdMapper.prepare_mapper(h5py_file, threedi_datasource)
mapper = IdMapper(h5py_file["mappings"]["id_map"])
assert mapper.get_by_code(TYPE_CODE_MAP["v2_channel"]).size == NODE_LENGTH
assert mapper.get_by_name("v2_channel").size == NODE_LENGTH