def test_end_period(self):
test_string = "$turnip 50 am"
self.assertEqual(extract_period(test_string), "AM")
def test_alan_bug(self):
alan_string = "$turnip am 89"
self.assertEqual(extract_period(alan_string), "AM")
def test_date_mid_period(self):
test_string = "$turnip 50 4/20 am"
self.assertEqual(extract_period(test_string), "AM")
def test_with_user(self):
test_user = "******"
self.assertEqual(extract_period(test_user), "AM")
def test_start_period(self):
test_string = "$turnip pm 50"
self.assertEqual(extract_period(test_string), "PM")