def test_tag_iterator(example_data_set):
"""
Tests the tag iterator.
"""
data_set = ASDFDataSet(example_data_set.filename)
expected_ids = ["AE.113A..BHE", "AE.113A..BHN", "AE.113A..BHZ", "TA.POKR..BHE", "TA.POKR..BHN", "TA.POKR..BHZ"]
for st, inv in data_set.itertag("raw_recording"):
for tr in st:
assert tr.id in expected_ids
expected_ids.remove(tr.id)
assert bool(
inv.select(
network=tr.stats.network,
station=tr.stats.station,
channel=tr.stats.channel,
location=tr.stats.location,
).networks
)
assert expected_ids == []
# It will only return matching tags.
count = 0
for _ in data_set.itertag("random"):
count += 1
assert count == 0
