修复生成节气错误 (#84)

Co-authored-by: yaoqiankun <yaoqiankun@datarc.cn>
LKI · Apr 9, 2022 · 111e92f · 111e92f
1 parent 4872c20
commit 111e92f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/chinese_calendar/utils.py b/chinese_calendar/utils.py
@@ -204,7 +204,10 @@ def get_solar_terms(start, end):
  delta = SOLAR_TERMS_DELTA.get((year, solar_term))
  if delta:
  day += delta
- result.append((datetime.date(year, month, day), solar_term.value[1]))
+ _date = datetime.date(year, month, day)
+ if _date < start or _date > end:
+ continue
+ result.append((_date, solar_term.value[1]))
  if month == 12:
  year, month = year + 1, 1
  else:

diff --git a/tests/test_solar_terms.py b/tests/test_solar_terms.py
@@ -657,3 +657,9 @@ def test_get_solar_terms(self):
  (datetime.date(2023, 12, 22), "冬至"),
  ]
  self.assertEqual(expected, actual)
+
+ start = datetime.date(2023, 1, 6)
+ end = datetime.date(2023, 1, 19)
+ actual = chinese_calendar.get_solar_terms(start, end)
+ expected = []
+ self.assertEqual(expected, actual)