def test_none(self): self.assertEqual(main.extract_cabin_floor(None), None)
def test_empty(self): self.assertEqual(main.extract_cabin_floor(""), None)
def test_floor_and_room_same_floor(self): self.assertEqual(main.extract_cabin_floor("C C45"), "C") self.assertEqual(main.extract_cabin_floor("C45 C"), "C")
def test_floor_and_room_different_floors(self): self.assertEqual(main.extract_cabin_floor("C D46"), "CD") self.assertEqual(main.extract_cabin_floor("C46 D"), "CD")
def test_multiple_room_different_floors(self): self.assertEqual(main.extract_cabin_floor("C45 D46"), "CD")
def test_multiple_room_same_floor(self): self.assertEqual(main.extract_cabin_floor("C45 C46"), "C")
def test_single_room(self): self.assertEqual(main.extract_cabin_floor("C45"), "C")
def test_multiple_floors_duplicated(self): self.assertEqual(main.extract_cabin_floor("C C D"), "CD")
def test_single_floor_duplicates(self): self.assertEqual(main.extract_cabin_floor("C C"), "C") self.assertEqual(main.extract_cabin_floor("C C C"), "C")