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