def test_decade(self): ct = get_time_unit_from_string('decade') self.assertEqual(ct, TimeUnit.DECADE)
def test_month(self): ct = get_time_unit_from_string('month') self.assertEqual(ct, TimeUnit.MONTH)
def test_year(self): ct = get_time_unit_from_string('year') self.assertEqual(ct, TimeUnit.YEAR)
def test_day(self): ct = get_time_unit_from_string('day') self.assertEqual(ct, TimeUnit.DAY)
def test_week(self): ct = get_time_unit_from_string('week') self.assertEqual(ct, TimeUnit.WEEK)
def test_hour(self): ct = get_time_unit_from_string('hour') self.assertEqual(ct, TimeUnit.HOUR)
def test_minute(self): ct = get_time_unit_from_string('minute') self.assertEqual(ct, TimeUnit.MINUTE)
def test_second(self): ct = get_time_unit_from_string('second') self.assertEqual(ct, TimeUnit.SECOND)
def test_none_or_empty(self): ct = get_time_unit_from_string(None) self.assertIsNone(ct) ct = get_time_unit_from_string('') self.assertIsNone(ct)
def test_invalid(self): with self.assertRaises(InvalidArgument): get_time_unit_from_string('foo')