def test_extract_annosets_with_ex_flatten_true():
annosets = list(
framenet.extract_annosets(SPLITS_DIRPATH,
False,
True, {},
flatten=True))
assert len(annosets) == 2
def test_extract_annosets_with_ft_flatten_false():
annosets = list(
framenet.extract_annosets(SPLITS_DIRPATH,
True,
False, {},
flatten=False))
assert len(annosets) == 2
assert len([item for sublist in annosets for item in sublist]) == 8
def test_extract_annosets_with_ft_with_ex_flatten_false():
annosets = list(
framenet.extract_annosets(SPLITS_DIRPATH,
with_fulltexts=True,
with_exemplars=True,
fe_dict={},
flatten=False))
assert len(annosets) == 4
assert len([item for sublist in annosets for item in sublist]) == 10