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