def test_get_lengths_works_with_words_and_characters(self): instance = IndexedTrueFalseInstance([[1, 2], [3, 1, 2]], True) assert instance.get_lengths() == { 'word_sequence_length': 2, 'word_character_length': 3 }
def test_get_lengths_returns_length_of_word_indices(self): instance = IndexedTrueFalseInstance([1, 2, 3, 4], True) assert instance.get_lengths() == {'word_sequence_length': 4}