Add tests for BeginningOfMonth and BeginningOfYear

M03ED · Nov 11, 2019 · cb9e20b · cb9e20b
1 parent 32373f1
commit cb9e20b
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/ptime_test.go b/ptime_test.go
@@ -336,6 +336,14 @@ func TestFirstLast(t *testing.T) {
 		)
 	}
 
+	if ti.BeginningOfMonth().Weekday() != Charshanbeh {
+		t.Error(
+			"For", "BeginningOfMonth().Weekday()",
+			"expected", Charshanbeh.String(),
+			"got", ti.BeginningOfMonth().Weekday(),
+		)
+	}
+
 	if ti.LastMonthDay().Weekday() != Panjshanbeh {
 		t.Error(
 			"For", "LastMonthDay().Weekday()",
@@ -352,6 +360,14 @@ func TestFirstLast(t *testing.T) {
 		)
 	}
 
+	if ti.BeginningOfYear().Weekday() != Shanbeh {
+		t.Error(
+			"For", "BeginningOfYear().Weekday()",
+			"expected", Shanbeh.String(),
+			"got", ti.BeginningOfYear().Weekday(),
+		)
+	}
+
 	if ti.LastYearDay().Weekday() != Shanbeh {
 		t.Error(
 			"For", "LastYearDay().Weekday()",