def testSerialize(self): date = datetime.date(2013, 2, 3) self.assertEqual(luigi.DateParameter().serialize(date), '2013-02-03') self.assertEqual(luigi.YearParameter().serialize(date), '2013') self.assertEqual(luigi.MonthParameter().serialize(date), '2013-02') dt = datetime.datetime(2013, 2, 3, 4, 5) self.assertEqual(luigi.DateHourParameter().serialize(dt), '2013-02-03T04')
def testYearWithIntervalParameter(self): p = luigi.YearParameter(config_path=dict(section="foo", name="bar"), start=datetime.date(2011, 1, 1), interval=5) self.assertEqual(datetime.date(2011, 1, 1), _value(p))
def testYearParameter(self): p = luigi.YearParameter(config_path=dict(section="foo", name="bar")) self.assertEqual(datetime.date(2015, 1, 1), _value(p))
class YearTask(luigi.Task): year = luigi.YearParameter()
def test_serialize(self): year = luigi.YearParameter().serialize(datetime.date(2015, 4, 3)) self.assertEqual(year, '2015')
def test_parse(self): year = luigi.YearParameter().parse('2015') self.assertEqual(year, datetime.date(2015, 1, 1))
class YearDefaultTask(luigi.task.Task): year = luigi.YearParameter(default=luigi.date_interval.Year(2015))