def test_month_date_range(self): """Test month_date_range returns correct month range.""" today = datetime.now() out = utils.month_date_range(today) start_month = today.replace(day=1, second=1, microsecond=1) end_month = start_month + relativedelta(months=+1) timeformat = "%Y%m%d" expected_string = f"{start_month.strftime(timeformat)}-{end_month.strftime(timeformat)}" self.assertEqual(out, expected_string)
def test_month_date_range(self): today = datetime.now() out = utils.month_date_range(today) start_month = today.replace(day=1, second=1, microsecond=1) end_month = start_month + relativedelta(months=+1) timeformat = '%Y%m%d' expected_string = '{}-{}'.format(start_month.strftime(timeformat), end_month.strftime(timeformat)) self.assertEqual(out, expected_string)
def _get_report_path(self, date_time): """ Return path of report files. Args: date_time (DateTime): The starting datetime object Returns: (String): "/prefix/report_name/YYYYMMDD-YYYYMMDD", example: "/my-prefix/my-report/19701101-19701201" """ report_date_range = utils.month_date_range(date_time) return f"{self.base_path}/{self.report_name}/{report_date_range}"
def _get_report_path(self, date_time): """ Return path of report files. Args: date_time (DateTime): The starting datetime object Returns: (String): "/prefix/report_name/YYYYMMDD-YYYYMMDD", example: "/my-prefix/my-report/19701101-19701201" """ report_date_range = utils.month_date_range(date_time) return "{}/{}/{}".format(self.report.get("S3Prefix"), self.report_name, report_date_range)