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