1 files changed, 285 insertions, 0 deletions
diff --git a/tests/datetime/test_start_end_of.py b/tests/datetime/test_start_end_of.py
new file mode 100644
index 0000000..597dbeb
--- /dev/null
+++ b/tests/datetime/test_start_end_of.py
@@ -0,0 +1,285 @@
+from __future__ import annotations
+
+import pytest
+
+import pendulum
+
+from tests.conftest import assert_datetime
+
+
+def test_start_of_second():
+    d = pendulum.now()
+    new = d.start_of("second")
+    assert isinstance(new, pendulum.DateTime)
+    assert_datetime(new, d.year, d.month, d.day, d.hour, d.minute, d.second, 0)
+
+
+def test_end_of_second():
+    d = pendulum.now()
+    new = d.end_of("second")
+    assert isinstance(new, pendulum.DateTime)
+    assert_datetime(new, d.year, d.month, d.day, d.hour, d.minute, d.second, 999999)
+
+
+def test_start_of_minute():
+    d = pendulum.now()
+    new = d.start_of("minute")
+    assert isinstance(new, pendulum.DateTime)
+    assert_datetime(new, d.year, d.month, d.day, d.hour, d.minute, 0, 0)
+
+
+def test_end_of_minute():
+    d = pendulum.now()
+    new = d.end_of("minute")
+    assert isinstance(new, pendulum.DateTime)
+    assert_datetime(new, d.year, d.month, d.day, d.hour, d.minute, 59, 999999)
+
+
+def test_start_of_hour():
+    d = pendulum.now()
+    new = d.start_of("hour")
+    assert isinstance(new, pendulum.DateTime)
+    assert_datetime(new, d.year, d.month, d.day, d.hour, 0, 0, 0)
+
+
+def test_end_of_hour():
+    d = pendulum.now()
+    new = d.end_of("hour")
+    assert isinstance(new, pendulum.DateTime)
+    assert_datetime(new, d.year, d.month, d.day, d.hour, 59, 59, 999999)
+
+
+def test_start_of_day():
+    d = pendulum.now()
+    new = d.start_of("day")
+    assert isinstance(new, pendulum.DateTime)
+    assert_datetime(new, d.year, d.month, d.day, 0, 0, 0, 0)
+
+
+def test_end_of_day():
+    d = pendulum.now()
+    new = d.end_of("day")
+    assert isinstance(new, pendulum.DateTime)
+    assert_datetime(new, d.year, d.month, d.day, 23, 59, 59, 999999)
+
+
+def test_start_of_month_is_fluid():
+    d = pendulum.now()
+    assert isinstance(d.start_of("month"), pendulum.DateTime)
+
+
+def test_start_of_month_from_now():
+    d = pendulum.now()
+    new = d.start_of("month")
+    assert_datetime(new, d.year, d.month, 1, 0, 0, 0, 0)
+
+
+def test_start_of_month_from_last_day():
+    d = pendulum.datetime(2000, 1, 31, 2, 3, 4)
+    new = d.start_of("month")
+    assert_datetime(new, 2000, 1, 1, 0, 0, 0, 0)
+
+
+def test_start_of_year_is_fluid():
+    d = pendulum.now()
+    new = d.start_of("year")
+    assert isinstance(new, pendulum.DateTime)
+
+
+def test_start_of_year_from_now():
+    d = pendulum.now()
+    new = d.start_of("year")
+    assert_datetime(new, d.year, 1, 1, 0, 0, 0, 0)
+
+
+def test_start_of_year_from_first_day():
+    d = pendulum.datetime(2000, 1, 1, 1, 1, 1)
+    new = d.start_of("year")
+    assert_datetime(new, 2000, 1, 1, 0, 0, 0, 0)
+
+
+def test_start_of_year_from_last_day():
+    d = pendulum.datetime(2000, 12, 31, 23, 59, 59)
+    new = d.start_of("year")
+    assert_datetime(new, 2000, 1, 1, 0, 0, 0, 0)
+
+
+def test_end_of_month_is_fluid():
+    d = pendulum.now()
+    assert isinstance(d.end_of("month"), pendulum.DateTime)
+
+
+def test_end_of_month():
+    d = pendulum.datetime(2000, 1, 1, 2, 3, 4).end_of("month")
+    new = d.end_of("month")
+    assert_datetime(new, 2000, 1, 31, 23, 59, 59)
+
+
+def test_end_of_month_from_last_day():
+    d = pendulum.datetime(2000, 1, 31, 2, 3, 4)
+    new = d.end_of("month")
+    assert_datetime(new, 2000, 1, 31, 23, 59, 59)
+
+
+def test_end_of_year_is_fluid():
+    d = pendulum.now()
+    assert isinstance(d.end_of("year"), pendulum.DateTime)
+
+
+def test_end_of_year_from_now():
+    d = pendulum.now().end_of("year")
+    new = d.end_of("year")
+    assert_datetime(new, d.year, 12, 31, 23, 59, 59, 999999)
+
+
+def test_end_of_year_from_first_day():
+    d = pendulum.datetime(2000, 1, 1, 1, 1, 1)
+    new = d.end_of("year")
+    assert_datetime(new, 2000, 12, 31, 23, 59, 59, 999999)
+
+
+def test_end_of_year_from_last_day():
+    d = pendulum.datetime(2000, 12, 31, 23, 59, 59, 999999)
+    new = d.end_of("year")
+    assert_datetime(new, 2000, 12, 31, 23, 59, 59, 999999)
+
+
+def test_start_of_decade_is_fluid():
+    d = pendulum.now()
+    assert isinstance(d.start_of("decade"), pendulum.DateTime)
+
+
+def test_start_of_decade_from_now():
+    d = pendulum.now()
+    new = d.start_of("decade")
+    assert_datetime(new, d.year - d.year % 10, 1, 1, 0, 0, 0, 0)
+
+
+def test_start_of_decade_from_first_day():
+    d = pendulum.datetime(2000, 1, 1, 1, 1, 1)
+    new = d.start_of("decade")
+    assert_datetime(new, 2000, 1, 1, 0, 0, 0, 0)
+
+
+def test_start_of_decade_from_last_day():
+    d = pendulum.datetime(2009, 12, 31, 23, 59, 59)
+    new = d.start_of("decade")
+    assert_datetime(new, 2000, 1, 1, 0, 0, 0, 0)
+
+
+def test_end_of_decade_is_fluid():
+    d = pendulum.now()
+    assert isinstance(d.end_of("decade"), pendulum.DateTime)
+
+
+def test_end_of_decade_from_now():
+    d = pendulum.now()
+    new = d.end_of("decade")
+    assert_datetime(new, d.year - d.year % 10 + 9, 12, 31, 23, 59, 59, 999999)
+
+
+def test_end_of_decade_from_first_day():
+    d = pendulum.datetime(2000, 1, 1, 1, 1, 1)
+    new = d.end_of("decade")
+    assert_datetime(new, 2009, 12, 31, 23, 59, 59, 999999)
+
+
+def test_end_of_decade_from_last_day():
+    d = pendulum.datetime(2009, 12, 31, 23, 59, 59, 999999)
+    new = d.end_of("decade")
+    assert_datetime(new, 2009, 12, 31, 23, 59, 59, 999999)
+
+
+def test_start_of_century_is_fluid():
+    d = pendulum.now()
+    assert isinstance(d.start_of("century"), pendulum.DateTime)
+
+
+def test_start_of_century_from_now():
+    d = pendulum.now()
+    new = d.start_of("century")
+    assert_datetime(new, d.year - d.year % 100 + 1, 1, 1, 0, 0, 0, 0)
+
+
+def test_start_of_century_from_first_day():
+    d = pendulum.datetime(2001, 1, 1, 1, 1, 1)
+    new = d.start_of("century")
+    assert_datetime(new, 2001, 1, 1, 0, 0, 0, 0)
+
+
+def test_start_of_century_from_last_day():
+    d = pendulum.datetime(2100, 12, 31, 23, 59, 59)
+    new = d.start_of("century")
+    assert_datetime(new, 2001, 1, 1, 0, 0, 0, 0)
+
+
+def test_end_of_century_is_fluid():
+    d = pendulum.now()
+    assert isinstance(d.end_of("century"), pendulum.DateTime)
+
+
+def test_end_of_century_from_now():
+    now = pendulum.now()
+    d = now.end_of("century")
+    assert_datetime(d, now.year - now.year % 100 + 100, 12, 31, 23, 59, 59, 999999)
+
+
+def test_end_of_century_from_first_day():
+    d = pendulum.datetime(2001, 1, 1, 1, 1, 1)
+    new = d.end_of("century")
+    assert_datetime(new, 2100, 12, 31, 23, 59, 59, 999999)
+
+
+def test_end_of_century_from_last_day():
+    d = pendulum.datetime(2100, 12, 31, 23, 59, 59, 999999)
+    new = d.end_of("century")
+    assert_datetime(new, 2100, 12, 31, 23, 59, 59, 999999)
+
+
+def test_average_is_fluid():
+    d = pendulum.now().average()
+    assert isinstance(d, pendulum.DateTime)
+
+
+def test_average_from_same():
+    d1 = pendulum.datetime(2000, 1, 31, 2, 3, 4)
+    d2 = pendulum.datetime(2000, 1, 31, 2, 3, 4).average(d1)
+    assert_datetime(d2, 2000, 1, 31, 2, 3, 4)
+
+
+def test_average_from_greater():
+    d1 = pendulum.datetime(2000, 1, 1, 1, 1, 1, tz="local")
+    d2 = pendulum.datetime(2009, 12, 31, 23, 59, 59, tz="local").average(d1)
+    assert_datetime(d2, 2004, 12, 31, 12, 30, 30)
+
+
+def test_average_from_lower():
+    d1 = pendulum.datetime(2009, 12, 31, 23, 59, 59, tz="local")
+    d2 = pendulum.datetime(2000, 1, 1, 1, 1, 1, tz="local").average(d1)
+    assert_datetime(d2, 2004, 12, 31, 12, 30, 30)
+
+
+def start_of_with_invalid_unit():
+    with pytest.raises(ValueError):
+        pendulum.now().start_of("invalid")
+
+
+def end_of_with_invalid_unit():
+    with pytest.raises(ValueError):
+        pendulum.now().end_of("invalid")
+
+
+def test_start_of_with_transition():
+    d = pendulum.datetime(2013, 10, 27, 23, 59, 59, tz="Europe/Paris")
+    assert d.offset == 3600
+    assert d.start_of("month").offset == 7200
+    assert d.start_of("day").offset == 7200
+    assert d.start_of("year").offset == 3600
+
+
+def test_end_of_with_transition():
+    d = pendulum.datetime(2013, 3, 31, tz="Europe/Paris")
+    assert d.offset == 3600
+    assert d.end_of("month").offset == 7200
+    assert d.end_of("day").offset == 7200
+    assert d.end_of("year").offset == 3600