def parse_time_room(self, tweet): """Get time and room number from a tweet using SUTime and tweet_utils""" extracted_time = self.sutime.parse(tweet) time_and_room = tweet_utils.get_time_and_room(tweet, extracted_time) return time_and_room
def test_get_time_and_room_with_only_time_present(self): tweet = "a test tweet with only a time present 2:05pm" result = tweet_utils.get_time_and_room(tweet, self.extracted_time) expected_output = {'room': [], 'date': ['2017-04-11T14:05']} self.assertEqual(result, expected_output)
def test_get_time_and_room_with_period_after_room_number(self): tweet = "a test tweet with a period after room num B112. 2:05pm" result = tweet_utils.get_time_and_room(tweet, self.extracted_time) expected_output = {'room': ['b112'], 'date': ['2017-04-11T14:05']} self.assertEqual(result, expected_output)
def test_get_time_and_room_with_only_room_present(self): tweet = "a test tweet with only a room number present A105+A106" result = tweet_utils.get_time_and_room(tweet, self.no_extracted_time) expected_output = {'room': ['a105+a106'], 'date': []} self.assertEqual(result, expected_output)
def test_get_time_and_room_correctly_returns_time_room_obj(self): tweet = "a test tweet B114 2:05pm" result = tweet_utils.get_time_and_room(tweet, self.extracted_time) expected_output = {'room': ['b114'], 'date': ['2017-04-11T14:05']} self.assertEqual(result, expected_output)