def test_split_sentences_and_assign_sentences_to_unit(self): long_sentence = 'In <strong>Unit 2</strong>, the temperature at the feed water nozzle of the RPV is 150 °C. In <strong>Unit 3</strong> the temperature at the feed water nozzle of the RPV is 91 °C and at the bottom of the RPV is 121 °C.' sentences = long_sentence.split('.') unit_parser = UnitSentenceParser() unit_sentence_assignments = {} current_unit = None for sentence in sentences: if unit_parser._is_unit_sentence(sentence): current_unit = unit_parser._parse_unit(sentence) if current_unit not in unit_sentence_assignments.keys(): unit_sentence_assignments[current_unit] = [] if current_unit: unit_sentence_assignments[current_unit].append(sentence) keys = unit_sentence_assignments.keys() keys.sort() self.assertEqual(keys, ['2', '3'])