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)