def test_check_runtime_range(self):
from luiti import luigi, TaskWeek, arrow
@luigi.check_runtime_range(
hour_num=[5, 6, 7, 8],
weekday_num=[1],
)
class CheckRuntimeRangeExampleWeek(TaskWeek):
root_dir = "/foobar"
def run(self):
return "data"
day_1 = arrow.get("2014-09-01 06:28") # valid
self.assertTrue(day_1)
def func(d1):
# overwrite arrow's method directly.
arrow.now = lambda: arrow.get(d1)
return CheckRuntimeRangeExampleWeek(d1).run()
self.assertEqual(func("2014-09-01 09:00"), False)
self.assertEqual(func("2014-09-02 06:28"), False)
self.assertEqual(func("2014-09-01 04:28"), False)
self.assertEqual(func("2014-09-01 05:00"), "data")
self.assertEqual(func("2014-09-01 06:28"), "data")
self.assertEqual(func("2014-09-01 08:59"), "data")
def test_check_runtime_range(self):
from luiti import luigi, TaskWeek, arrow
@luigi.check_runtime_range(hour_num=[5, 6, 7, 8], weekday_num=[1], )
class CheckRuntimeRangeExampleWeek(TaskWeek):
root_dir = "/foobar"
def run(self):
return "data"
day_1 = arrow.get("2014-09-01 06:28") # valid
self.assertTrue(day_1)
def func(d1):
# overwrite arrow's method directly.
arrow.now = lambda: arrow.get(d1)
return CheckRuntimeRangeExampleWeek(d1).run()
self.assertEqual(func("2014-09-01 09:00"), False)
self.assertEqual(func("2014-09-02 06:28"), False)
self.assertEqual(func("2014-09-01 04:28"), False)
self.assertEqual(func("2014-09-01 05:00"), "data")
self.assertEqual(func("2014-09-01 06:28"), "data")
self.assertEqual(func("2014-09-01 08:59"), "data")
def func(d1):
# overwrite arrow's method directly.
arrow.now = lambda: arrow.get(d1)
return CheckRuntimeRangeExampleWeek(d1).run()
def func(d1):
# overwrite arrow's method directly.
arrow.now = lambda: arrow.get(d1)
return CheckRuntimeRangeExampleWeek(d1).run()
