def main(args): if args.verbose: log_level = logging.DEBUG else: log_level = logging.INFO logging.basicConfig(level=log_level, format=LOG_FORMAT) date_step = cli.get_interval(args) _do_cross_table_consistency_check(args.project, args.start_date, args.end_date, date_step)
def test_get_interval_raises_error_when_no_interval_is_specified(self): mock_args = mock.Mock(interval_days=None, interval_months=None) with self.assertRaises(ValueError): cli.get_interval(mock_args)
def test_get_interval_when_months_specified(self): interval_months = relativedelta.relativedelta(months=3) mock_args = mock.Mock( interval_days=None, interval_months=relativedelta.relativedelta(months=3)) self.assertEqual(interval_months, cli.get_interval(mock_args))