diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index f35df46..5f7b9f6 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -12,7 +12,6 @@ jobs: - uses: psf/black@stable with: options: "--check" - pylint unit_of_time test: name: run tests runs-on: ubuntu-latest diff --git a/timetest.py b/timetest.py index 48a655e..c871973 100644 --- a/timetest.py +++ b/timetest.py @@ -110,6 +110,8 @@ def test_to_int(self): self.assertIn(dt, tu) self.assertIn((dt, dt), tu) self.assertIn((tu.first_date, tu.last_date), tu) + with self.assertRaises(TypeError): + self.assertIn((1958, 2019), tu) with self.assertRaises(TypeError): self.assertIn(1425, tu) self.assertIn(dt, list(tu)) @@ -151,8 +153,12 @@ def test_to_int(self): self.assertEqual(tu.previous.previous.previous, 3 << tu) self.assertLess(tu.last_date, tu.next.first_date) self.assertLess(tu.previous.last_date, tu.first_date) - self.assertEqual((tu.next.first_date - tu.last_date), timedelta(days=1)) - self.assertEqual((tu.first_date - tu.previous.last_date), timedelta(days=1)) + self.assertEqual( + (tu.next.first_date - tu.last_date), timedelta(days=1) + ) + self.assertEqual( + (tu.first_date - tu.previous.last_date), timedelta(days=1) + ) def test_hierarchy(self): """ diff --git a/unit_of_time/__init__.py b/unit_of_time/__init__.py index 3ce1503..7421d21 100644 --- a/unit_of_time/__init__.py +++ b/unit_of_time/__init__.py @@ -36,7 +36,9 @@ def unit_register(cls): result = TimeunitKindMeta._registered if result is None: result = { - k.kind_int: k for k in TimeunitKindMeta._pre_registered if k.kind_int is not None + k.kind_int: k + for k in TimeunitKindMeta._pre_registered + if k.kind_int is not None } TimeunitKindMeta._registered = result return result