Skip to content

Commit

Permalink
🐛 fix
Browse files Browse the repository at this point in the history
  • Loading branch information
yezz123 committed Jul 1, 2023
1 parent ef063e6 commit 0d162d1
Showing 1 changed file with 21 additions and 18 deletions.
39 changes: 21 additions & 18 deletions tests/internal/test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,11 +36,24 @@
)


@pytest.fixture
def sample_datetime():
return datetime(2023, 5, 20, 12, 0, 0)


@pytest.fixture
def sample_datetime_with_tz():
tz = pytz.timezone("America/New_York")
return tz.localize(datetime(2023, 5, 20, 12, 0, 0))


@freeze_time("2023-07-01 12:00:00")
def test_util_get_now():
assert isinstance(get_now(), date.datetime)
assert get_now().tzinfo == date.timezone.utc


@freeze_time("2023-07-01 12:00:00")
def test_util_get_now_ts():
assert isinstance(get_now_ts(), float)

Expand All @@ -53,17 +66,7 @@ def test_util_get_uuid():
utc = timezone("UTC")


@pytest.fixture
def sample_datetime():
return datetime(2023, 5, 20, 12, 0, 0)


@pytest.fixture
def sample_datetime_with_tz():
tz = pytz.timezone("America/New_York")
return tz.localize(datetime(2023, 5, 20, 12, 0, 0))


@freeze_time("2023-07-01 12:00:00")
def test_time_diff():
dt1 = datetime(2023, 5, 20, 12, 0, 0)
dt2 = datetime(2023, 5, 19, 12, 0, 0)
Expand Down Expand Up @@ -121,17 +124,17 @@ def test_months_after(sample_datetime):
assert months_after(sample_datetime, months=1) == expected_result


def test_years_ago():
dt = datetime(2023, 5, 20, 12, 0, 0)
def test_years_ago(sample_datetime):
expected_result = datetime(2022, 5, 20, 12, 0, 0)
assert years_ago(dt, years=1) == expected_result
assert years_ago(sample_datetime, years=1) == expected_result


def test_days_after(sample_datetime):
expected_result = datetime(2023, 5, 21, 12, 0, 0)
assert days_after(sample_datetime, days=1) == expected_result


@freeze_time("2023-07-01 12:00:00")
def test_is_today():
assert is_today(datetime.now()) is True

Expand All @@ -140,7 +143,7 @@ def test_is_yesterday(sample_datetime):
assert is_yesterday(sample_datetime) is False


@freeze_time("2023-05-20 12:00:00")
@freeze_time("2023-07-22 12:00:00")
def test_is_tomorrow(sample_datetime):
assert is_tomorrow(sample_datetime) is False

Expand All @@ -161,17 +164,17 @@ def test_tz_from_iso():

def test_start_of_week():
dt = datetime(2023, 5, 20, 12, 0, 0)
expected_result = datetime(2023, 5, 14, 0, 0, 0)
expected_result = datetime(2023, 5, 15, 12, 0, 0)
assert start_of_week(dt) == expected_result


def test_end_of_week():
dt = datetime(2023, 5, 20, 12, 0, 0)
expected_result = datetime(2023, 5, 20, 23, 59, 59, 999999)
expected_result = datetime(2023, 5, 21, 12, 0)
assert end_of_week(dt) == expected_result


def test_end_of_last_week():
dt = datetime(2023, 5, 20, 12, 0, 0)
expected_result = datetime(2023, 5, 13, 23, 59, 59, 999999)
expected_result = datetime(2023, 5, 14, 12, 0)
assert end_of_last_week(dt) == expected_result

0 comments on commit 0d162d1

Please sign in to comment.