def test_get_lengths_returns_correct_background_length(self):
instance = IndexedBackgroundInstance(self.base_instance,
self.background_instances)
assert instance.get_lengths() == {
'word_sequence_length': 3,
'background_sentences': 2
}
def test_get_lengths_returns_max_of_background_and_word_indices(self):
instance = IndexedBackgroundInstance(self.base_instance,
self.background_instances)
assert instance.get_lengths()['word_sequence_length'] == 3