Skip to content

Commit

Permalink
Merge pull request #87 from edx/added_blank_lines
Browse files Browse the repository at this point in the history
fix: fix tests by adding blank lines
  • Loading branch information
usama101 authored Aug 9, 2023
2 parents 758764f + 2c6b5cd commit 9a24a47
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 0 deletions.
1 change: 1 addition & 0 deletions edx_prefectutils/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,7 @@ def generate_dates(start_date: str, end_date: str, date_format: str = "%Y%m%d"):

return [date.strftime(date_format) for date in dates]


@task
def generate_month_start_dates(start_date: str, end_date: str, date_format: str = "%Y-%m-%d"):
"""
Expand Down
2 changes: 2 additions & 0 deletions tests/test_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ def test_generate_dates():
assert state.is_successful()
assert state.result[task].result == ['20200101', '20200102', '20200103', '20200104']


def test_generate_month_start_dates():
with Flow("test") as f:
task = common.generate_month_start_dates(
Expand All @@ -29,6 +30,7 @@ def test_generate_month_start_dates():
assert state.is_successful()
assert state.result[task].result == ['2023-01-01', '2023-02-01', '2023-03-01', '2023-04-01', '2023-05-01']


def test_get_unzipped_cartesian_product():
with Flow("test") as f:
task = common.get_unzipped_cartesian_product(
Expand Down

0 comments on commit 9a24a47

Please sign in to comment.