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))
