def test_enum_multiple_error():
grp = AFMGroup(data_dir / "map2x2_extracted.jpk-force-map")
grp += load_data(data_dir / "spot3-0192.jpk-force")
with pytest.raises(ValueError, match="Multiple curves"):
grp.get_enum(0)
def test_enum_not_found_error():
grp = AFMGroup(data_dir / "map2x2_extracted.jpk-force-map")
with pytest.raises(KeyError, match="Could not find"):
grp.get_enum(100)
def test_enum_index():
grp = AFMGroup(data_dir / "map2x2_extracted.jpk-force-map")
afmd = grp.get_enum(2)
assert afmd.enum == 2
assert grp.index(afmd) == 2