增加自动化测试

.flake8

@@ -0,0 +1,2 @@
+[flake8]
+max-line-length = 100

.github/workflows/python-app.yml

@@ -0,0 +1,36 @@
+# This workflow will install Python dependencies, run tests and lint with a single version of Python
+# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions
+
+name: Python application
+
+on:
+  push:
+    branches: [ master ]
+  pull_request:
+    branches: [ master ]
+
+jobs:
+  build:
+
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Python 3.7
+      uses: actions/setup-python@v2
+      with:
+        python-version: 3.7
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
+    - name: Install dependencies for testing
+      run: |
+        if [ -f requirements-test.txt ]; then pip install -r requirements-test.txt; fi
+    - name: Lint with flake8
+      run: |
+        # stop the build if there are Python syntax errors or undefined names
+        flake8 . --count --show-source --statistics
+    - name: Test with pytest
+      run: |
+        python -m pytest

README.md

@@ -40,27 +40,27 @@ pip install zhdate --upgrade
 ```python
 from zhdate import ZhDate
 
-date1 = ZhDate(2010, 1, 1) # 新建农历 2010年正月初一 的日期对象
+date1 = ZhDate(2010, 1, 1)  # 新建农历 2010年正月初一 的日期对象
 print(date1)  # 直接返回农历日期字符串
 dt_date1 = date1.to_datetime() # 农历转换成阳历日期 datetime 类型
 
 dt_date2 = datetime(2010, 2, 6)
-date2 = ZhDate.from_datetime(dt_date2) # 从阳历日期转换成农历日期对象
+date2 = ZhDate.from_datetime(dt_date2)  # 从阳历日期转换成农历日期对象
 
-date3 = ZhDate(2020, 4, 30, leap_month=True) # 新建农历 2020年闰4月30日
+date3 = ZhDate(2020, 4, 30, leap_month=True)  # 新建农历 2020年闰4月30日
 print(date3.to_datetime())
 
 # 支持比较
 if ZhDate(2019, 1, 1) == ZhDate.from_datetime(datetime(2019, 2, 5)):
     pass
 
 # 减法支持
-new_zhdate = ZhDate(2019, 1, 1) - 30  #减整数，得到差额天数的新农历对象
-new_zhdate2 = ZhDate(2019, 1, 1) - ZhDate(2018, 1, 1) #两个zhdate对象相减得到两个农历日期的差额
-new_zhdate3 = ZhDate(2019, 1, 1) - datetime(2019, 1, 1) # 减去阳历日期，得到农历日期和阳历日期之间的天数差额
+new_zhdate = ZhDate(2019, 1, 1) - 30  # 减整数，得到差额天数的新农历对象
+delta1 = ZhDate(2019, 1, 1) - ZhDate(2018, 1, 1) # 两个zhdate对象相减得到两个农历日期的差额
+delta2 = ZhDate(2019, 1, 1) - datetime(2019, 1, 1) # 减去阳历日期，得到农历日期和阳历日期之间的天数差额
 
 # 加法支持
-new_zhdate4 = ZhDate(2019, 1, 1) + 30 # 加整数返回相隔天数以后的新农历对象
+new_zhdate4 = ZhDate(2019, 1, 1) + 30  # 加整数返回相隔天数以后的新农历对象
 
 # 中文输出
 new_zhdate5 = ZhDate(2019, 1, 1)

requirements-test.txt

@@ -0,0 +1,3 @@
+flake8==3.9.0
+importlib-metadata==4.13.0
+pytest==6.0.1

sample_zhdate.py

@@ -1,28 +1,30 @@
+# -*- coding=utf-8 -*-
+
 from zhdate import ZhDate
 from datetime import datetime
 
 if __name__ == '__main__':
-    date1 = ZhDate(2010, 1, 1) # 新建农历 2010年正月初一 的日期对象
+    date1 = ZhDate(2010, 1, 1)  # 新建农历 2010年正月初一 的日期对象
     # print(date1)  # 直接返回农历日期字符串
-    dt_date1 = date1.to_datetime() # 农历转换成阳历日期 datetime 类型
+    dt_date1 = date1.to_datetime()  # 农历转换成阳历日期 datetime 类型
 
-    dt_date2 = datetime(2010, 2, 6) 
-    date2 = ZhDate.from_datetime(dt_date2) # 从阳历日期转换成农历日期对象
+    dt_date2 = datetime(2010, 2, 6)
+    date2 = ZhDate.from_datetime(dt_date2)  # 从阳历日期转换成农历日期对象
 
-    date3 = ZhDate(2020, 4, 29, leap_month=True) # 新建农历 2020年闰4月30日
+    date3 = ZhDate(2020, 4, 29, leap_month=True)  # 新建农历 2020年闰4月30日
     # print(date3.to_datetime())
 
     # 支持比较
     if ZhDate(2019, 1, 1) == ZhDate.from_datetime(datetime(2019, 2, 5)):
         pass
 
     # 减法支持
-    new_zhdate = ZhDate(2019, 1, 1) - 30  #减整数，得到差额天数的新农历对象
-    new_zhdate2 = ZhDate(2019, 1, 1) - ZhDate(2018, 1, 1) #两个zhdate对象相减得到两个农历日期的差额
-    new_zhdate3 = ZhDate(2019, 1, 1) - datetime(2019, 1, 1) # 减去阳历日期，得到农历日期和阳历日期之间的天数差额
+    new_zhdate = ZhDate(2019, 1, 1) - 30  # 减整数，得到差额天数的新农历对象
+    delta1 = ZhDate(2019, 1, 1) - ZhDate(2018, 1, 1)  # 两个zhdate对象相减得到两个农历日期的差额
+    delta2 = ZhDate(2019, 1, 1) - datetime(2019, 1, 1)  # 减去阳历日期，得到农历日期和阳历日期之间的天数差额
 
     # 加法支持
-    new_zhdate4 = ZhDate(2019, 1, 1) + 30 # 加整数返回相隔天数以后的新农历对象
+    new_zhdate4 = ZhDate(2019, 1, 1) + 30  # 加整数返回相隔天数以后的新农历对象
 
     print(ZhDate(1900, 9, 1, False).chinese())
     print(ZhDate.today().chinese())

setup.py

@@ -1,3 +1,5 @@
+# -*- coding=utf-8 -*-
+
 import setuptools
 
 '''
@@ -6,7 +8,7 @@
 LastEditors  : PandaWithBeard
 LastEditTime : 2023-01-21 02:30:41
 FilePath     : /zhdate/setup.py
-Description  : 
+Description  :
 '''
 
 with open('README.md', 'r') as file:
@@ -21,12 +23,10 @@
     long_description=long_description,
     long_description_content_type="text/markdown",
     url='https://github.com/CutePandaSh/zhdate',
-    packages=setuptools.find_packages(),
+    packages=setuptools.find_packages(exclude=['tests']),
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)",
         "Operating System :: OS Independent",
     ],
 )
-
-

test_zhdate.py → tests/test_zhdate.py

@@ -1,3 +1,5 @@
+# -*- coding=utf-8 -*-
+
 import unittest
 from datetime import datetime
 from zhdate import ZhDate
@@ -95,4 +97,4 @@ def test_validate(self):
         ]
 
         for case in test_cases:
-            self.assertEqual(ZhDate.validate(*case[0]), case[1])
+            self.assertEqual(ZhDate.validate(*case[0]), case[1])

zhdate/__init__.py

@@ -1,13 +1,14 @@
-'''
--*- coding: utf-8 -*-
+# -*- coding: utf-8 -*-
+
+"""
 File: zhdate.py
 File Created: Sunday, 17th February 2019 4:58:02 pm
 Author: Wang, Yi ([email protected])
-'''
-'''
+-----
 changed: Saturday, 21st January 2023
 by: Eilles Wan ([email protected])
-'''
+"""
+
 from datetime import datetime, timedelta
 from itertools import accumulate
 from .constants import CHINESENEWYEAR, CHINESEYEARCODE
@@ -55,17 +56,19 @@ def from_datetime(dt):
             ZhDate -- 生成的农历日期对象
         """
         lunar_year = dt.year
-        # 如果还没有到农历正月初一 农历年份减去1
-        lunar_year -= (datetime.strptime(CHINESENEWYEAR[lunar_year-1900], '%Y%m%d') - dt).total_seconds() > 0
         # 当时农历新年时的日期对象
         newyear_dt = datetime.strptime(CHINESENEWYEAR[lunar_year-1900], '%Y%m%d')
+        # 如果还没有到农历正月初一 农历年份减去1
+        lunar_year -= (newyear_dt - dt).total_seconds() > 0
         # 查询日期距离当年的春节差了多久
         days_passed = (dt - newyear_dt).days
         # 被查询日期的年份码
         year_code = CHINESEYEARCODE[lunar_year - 1900]
         # 取得本年的月份列表
         month_days = ZhDate.decode(year_code)
 
+        month = 0
+        lunar_day = 0
         for pos, days in enumerate(accumulate(month_days)):
             if days_passed + 1 <= days:
                 month = pos + 1
@@ -94,13 +97,13 @@ def __days_passed(self):
             int -- 差值天数
         """
         month_days = ZhDate.decode(self.year_code)
-        #当前农历年的闰月，为0表示无润叶
-        month_leap =  self.year_code & 0xf
+        # 当前农历年的闰月，为0表示无润叶
+        month_leap = self.year_code & 0xf
 
-        #当年无闰月，或者有闰月但是当前月小于闰月
+        # 当年无闰月，或者有闰月但是当前月小于闰月
         if (month_leap == 0) or (self.lunar_month < month_leap):
             days_passed_month = sum(month_days[:self.lunar_month - 1])
-        #当前不是闰月，并且当前月份和闰月相同
+        # 当前不是闰月，并且当前月份和闰月相同
         elif (not self.leap_month) and (self.lunar_month == month_leap):
             days_passed_month = sum(month_days[:self.lunar_month - 1])
         else:
@@ -157,7 +160,11 @@ def __str__(self):
         Returns:
             str -- 标准格式农历日期字符串
         """
-        return "农历{}年{}{}月{}日".format(self.lunar_year,"闰" if self.leap_month else "",self.lunar_month,self.lunar_day)
+        return "农历{}年{}{}月{}日".format(
+            self.lunar_year, "闰" if self.leap_month else "",
+            self.lunar_month,
+            self.lunar_day
+        )
 
     def __repr__(self):
         return self.__str__()
@@ -194,7 +201,7 @@ def __sub__(self, another):
     def __tiandi(anum):
         tian = '甲乙丙丁戊己庚辛壬癸'
         di = '子丑寅卯辰巳午未申酉戌亥'
-        return '{}{}'.format(tian[anum % 10],di[anum % 12])
+        return '{}{}'.format(tian[anum % 10], di[anum % 12])
 
     @staticmethod
     def validate(year, month, day, leap):
@@ -217,15 +224,15 @@ def validate(year, month, day, leap):
 
         # 有闰月标志
         if leap:
-            if (year_code & 0xf) != month: # 年度闰月和校验闰月不一致的话，返回校验失败
+            if (year_code & 0xf) != month:  # 年度闰月和校验闰月不一致的话，返回校验失败
                 return False
             elif day == 30:  # 如果日期是30的话，直接返回年度代码首位是否为1，即闰月是否为大月
                 return (year_code >> 16) == 1
-            else: # 年度闰月和当前月份相同，日期不为30的情况，返回通过
+            else:  # 年度闰月和当前月份相同，日期不为30的情况，返回通过
                 return True
-        elif day <= 29: # 非闰月，并且日期小于等于29，返回通过
+        elif day <= 29:  # 非闰月，并且日期小于等于29，返回通过
             return True
-        else: # 非闰月日期为30，返回年度代码中的月份位是否为1，即是否为大月
+        else:  # 非闰月日期为30，返回年度代码中的月份位是否为1，即是否为大月
             return ((year_code >> (12 - month) + 4) & 1) == 1
 
     @staticmethod

zhdate/constants.py

@@ -1,9 +1,10 @@
-'''
--*- coding: utf-8 -*-
+# -*- coding: utf-8 -*-
+
+"""
 File: constant.py
 File Created: Saturday, 21st January 2023 01:42 am
 Author: Eilles Wan ([email protected])
-'''
+"""
 
 CHINESEYEARCODE = [
         19416,
@@ -41,7 +42,7 @@
 最后四位：表示闰月的月份，0表示当年无闰月
 
 前四位和最后四位应该结合使用，如果最后四位为0，则不考虑前四位
-例： 
+例：
 1901年代码为 19168，转成二进制为 0b100101011100000, 最后四位为0，当年无闰月，月份数据为 010010101110 分别代表12月的大小情况
 1903年代码为 21717，转成二进制为 0b101010011010101，最后四位为5，当年为闰五月，首四位为0，闰月为29天，月份数据为 010101001101 分别代表12月的大小情况
 
@@ -93,4 +94,3 @@
 '''
 从1900年，至2100年每年的农历春节的公历日期
 '''
-