def test_Example_from_dict_with_sent_start(annots): vocab = Vocab() predicted = Doc(vocab, words=annots["words"]) example = Example.from_dict(predicted, annots) assert len(list(example.reference.sents)) == 2 for i, token in enumerate(example.reference): if to_ternary_int(annots["sent_starts"][i]) == 1: assert token.is_sent_start is True elif to_ternary_int(annots["sent_starts"][i]) == 0: assert token.is_sent_start is None else: assert token.is_sent_start is False
def test_to_ternary_int(): assert to_ternary_int(True) == 1 assert to_ternary_int(None) == 0 assert to_ternary_int(False) == -1 assert to_ternary_int(1) == 1 assert to_ternary_int(1.0) == 1 assert to_ternary_int(0) == 0 assert to_ternary_int(0.0) == 0 assert to_ternary_int(-1) == -1 assert to_ternary_int(5) == -1 assert to_ternary_int(-10) == -1 assert to_ternary_int("string") == -1 assert to_ternary_int([0, "string"]) == -1