From 125e13f1db5c4b10ee1fa05d5747232bd9191072 Mon Sep 17 00:00:00 2001
From: Jacek Migdal <jacek@migdal.pl>
Date: Mon, 23 Sep 2024 17:15:19 +0200
Subject: [PATCH] Better ISO 8601 parsing (#26)

Adds support for YYYY-MM
---
 iso8601.go      |  3 +++
 iso8601_test.go | 12 ++++++++++++
 2 files changed, 15 insertions(+)

diff --git a/iso8601.go b/iso8601.go
index fd4c577..f4f4e8c 100644
--- a/iso8601.go
+++ b/iso8601.go
@@ -215,6 +215,9 @@ parse:
 			Y = c
 			M = 1
 			d = 1
+		case month:
+			M = c
+			d = 1
 		case day:
 			d = c
 		case hour:
diff --git a/iso8601_test.go b/iso8601_test.go
index 96a3347..1336014 100644
--- a/iso8601_test.go
+++ b/iso8601_test.go
@@ -251,6 +251,18 @@ var cases = []TestCase{
 		MilliSecond: 0,
 		Zone:        0,
 	},
+	{
+		Using: "2017-02",
+		Year:  2017, Month: 2, Day: 1, Hour: 0, Minute: 0, Second: 0,
+		MilliSecond: 0,
+		Zone:        0,
+	},
+	{
+		Using: "2017-02-16",
+		Year:  2017, Month: 2, Day: 16, Hour: 0, Minute: 0, Second: 0,
+		MilliSecond: 0,
+		Zone:        0,
+	},
 
 	// Invalid Parse Test Cases
 	{