v0.0.585

timeext/time_test.go

@@ -227,3 +227,173 @@ func TestDaysInMonth_FebruaryNonLeapYear(t *testing.T) {
 		t.Errorf("Expected %d but got %d", expected, result)
 	}
 }
+
+func TestAvg_MultipleValues(t *testing.T) {
+	t1 := time.Date(2022, 1, 1, 0, 0, 0, 0, time.UTC)
+	t2 := time.Date(2022, 1, 3, 0, 0, 0, 0, time.UTC)
+	t3 := time.Date(2022, 1, 5, 0, 0, 0, 0, time.UTC)
+
+	// Average should be January 3, 2022 (middle date)
+	expected := time.Date(2022, 1, 3, 0, 0, 0, 0, time.UTC)
+	result := Avg(t1, t2, t3)
+
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestAvg_TwoValues(t *testing.T) {
+	t1 := time.Date(2022, 1, 1, 0, 0, 0, 0, time.UTC)
+	t2 := time.Date(2022, 1, 3, 0, 0, 0, 0, time.UTC)
+
+	expected := time.Date(2022, 1, 2, 0, 0, 0, 0, time.UTC)
+	result := Avg(t1, t2)
+
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestAvg_EmptySlice(t *testing.T) {
+	result := Avg()
+	expected := time.Time{}
+
+	if !result.Equal(expected) {
+		t.Errorf("Expected zero time but got %v", result)
+	}
+}
+
+func TestMedian_OddNumberOfValues(t *testing.T) {
+	t1 := time.Date(2022, 1, 1, 0, 0, 0, 0, time.UTC)
+	t2 := time.Date(2022, 1, 3, 0, 0, 0, 0, time.UTC)
+	t3 := time.Date(2022, 1, 5, 0, 0, 0, 0, time.UTC)
+
+	// Median should be the middle date
+	expected := t2
+	result := Median(t1, t2, t3)
+
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestMedian_EvenNumberOfValues(t *testing.T) {
+	t1 := time.Date(2022, 1, 1, 0, 0, 0, 0, time.UTC)
+	t2 := time.Date(2022, 1, 2, 0, 0, 0, 0, time.UTC)
+	t3 := time.Date(2022, 1, 3, 0, 0, 0, 0, time.UTC)
+	t4 := time.Date(2022, 1, 4, 0, 0, 0, 0, time.UTC)
+
+	// Median for even number of values should be average of middle two
+	expected := time.Date(2022, 1, 2, 12, 0, 0, 0, time.UTC)
+	result := Median(t1, t2, t3, t4)
+
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestMedian_UnsortedValues(t *testing.T) {
+	t1 := time.Date(2022, 1, 5, 0, 0, 0, 0, time.UTC)
+	t2 := time.Date(2022, 1, 1, 0, 0, 0, 0, time.UTC)
+	t3 := time.Date(2022, 1, 3, 0, 0, 0, 0, time.UTC)
+
+	// Median should correctly sort values first
+	expected := t3
+	result := Median(t1, t2, t3)
+
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestMedian_EmptySlice(t *testing.T) {
+	result := Median()
+	expected := time.Time{}
+
+	if !result.Equal(expected) {
+		t.Errorf("Expected zero time but got %v", result)
+	}
+}
+
+func TestTimeToDatePart(t *testing.T) {
+	tz := TimezoneBerlin
+	tm := time.Date(2022, 1, 1, 13, 14, 15, 0, tz)
+	expected := time.Date(2022, 1, 1, 0, 0, 0, 0, tz)
+	result := TimeToDatePart(tm, tz)
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestTimeToWeekStart(t *testing.T) {
+	tz := TimezoneBerlin
+	// January 5, 2022 was a Wednesday
+	tm := time.Date(2022, 1, 5, 13, 14, 15, 0, tz)
+	// Should return Monday, January 3, 2022
+	expected := time.Date(2022, 1, 3, 0, 0, 0, 0, tz)
+	result := TimeToWeekStart(tm, tz)
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestTimeToWeekStart_WhenMonday(t *testing.T) {
+	tz := TimezoneBerlin
+	// January 3, 2022 was a Monday
+	tm := time.Date(2022, 1, 3, 13, 14, 15, 0, tz)
+	expected := time.Date(2022, 1, 3, 0, 0, 0, 0, tz)
+	result := TimeToWeekStart(tm, tz)
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestTimeToMonthStart(t *testing.T) {
+	tz := TimezoneBerlin
+	tm := time.Date(2022, 1, 15, 13, 14, 15, 0, tz)
+	expected := time.Date(2022, 1, 1, 0, 0, 0, 0, tz)
+	result := TimeToMonthStart(tm, tz)
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestTimeToMonthEnd(t *testing.T) {
+	tz := TimezoneBerlin
+	tm := time.Date(2022, 1, 15, 13, 14, 15, 0, tz)
+	expected := time.Date(2022, 2, 1, 0, 0, 0, 0, tz).Add(-1)
+	result := TimeToMonthEnd(tm, tz)
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestTimeToYearStart(t *testing.T) {
+	tz := TimezoneBerlin
+	tm := time.Date(2022, 5, 15, 13, 14, 15, 0, tz)
+	expected := time.Date(2022, 1, 1, 0, 0, 0, 0, tz)
+	result := TimeToYearStart(tm, tz)
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestTimeToYearEnd(t *testing.T) {
+	tz := TimezoneBerlin
+	tm := time.Date(2022, 5, 15, 13, 14, 15, 0, tz)
+	expected := time.Date(2023, 1, 1, 0, 0, 0, 0, tz).Add(-1)
+	result := TimeToYearEnd(tm, tz)
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}
+
+func TestTimeToNextYearStart(t *testing.T) {
+	tz := TimezoneBerlin
+	tm := time.Date(2022, 5, 15, 13, 14, 15, 0, tz)
+	expected := time.Date(2023, 1, 1, 0, 0, 0, 0, tz)
+	result := TimeToNextYearStart(tm, tz)
+	if !result.Equal(expected) {
+		t.Errorf("Expected %v but got %v", expected, result)
+	}
+}